Skip to content

Commit dbfeffe

Browse files
committed
include VLB models in README
1 parent b992106 commit dbfeffe

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

README.md

+16
Original file line numberDiff line numberDiff line change
@@ -128,3 +128,19 @@ MODEL_FLAGS="--image_size 256 --num_channels 128 --num_res_blocks 2 --num_heads
128128
DIFFUSION_FLAGS="--diffusion_steps 1000 --noise_schedule linear --rescale_learned_sigmas False --rescale_timesteps False --use_scale_shift_norm False"
129129
TRAIN_FLAGS="--lr 2e-5 --batch_size 128"
130130
```
131+
132+
Unconditional ImageNet-64 with the `L_vlb` objective and cosine noise schedule [[checkpoint](https://openaipublic.blob.core.windows.net/diffusion/march-2021/imagenet64_uncond_vlb_100M_1500K.pt)]:
133+
134+
```bash
135+
MODEL_FLAGS="--image_size 64 --num_channels 128 --num_res_blocks 3 --learn_sigma True"
136+
DIFFUSION_FLAGS="--diffusion_steps 4000 --noise_schedule cosine"
137+
TRAIN_FLAGS="--lr 1e-4 --batch_size 128 --schedule_sampler loss-second-moment"
138+
```
139+
140+
Unconditional CIFAR-10 with the `L_vlb` objective and cosine noise schedule [[checkpoint](https://openaipublic.blob.core.windows.net/diffusion/march-2021/cifar10_uncond_vlb_50M_500K.pt)]:
141+
142+
```bash
143+
MODEL_FLAGS="--image_size 32 --num_channels 128 --num_res_blocks 3 --learn_sigma True --dropout 0.3"
144+
DIFFUSION_FLAGS="--diffusion_steps 4000 --noise_schedule cosine"
145+
TRAIN_FLAGS="--lr 1e-4 --batch_size 128 --schedule_sampler loss-second-moment"
146+
```

0 commit comments

Comments
 (0)