Skip to content

Commit 783b674

Browse files
authored
Update README.md
1 parent 2e66f3f commit 783b674

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ Here are some changes we experiment with, and how to set them in the flags:
3838

3939
* **Learned sigmas:** add `--learn_sigma True` to `MODEL_FLAGS`
4040
* **Cosine schedule:** change `--noise_schedule linear` to `--noise_schedule cosine`
41-
* **Reweighted VLB:** add `--use_kl True` to `DIFFUSION_FLAGS` and add `--schedule_sampler loss-second-moment` to `TRAIN_FLAGS`.
41+
* **Importance-sampled VLB:** add `--use_kl True` to `DIFFUSION_FLAGS` and add `--schedule_sampler loss-second-moment` to `TRAIN_FLAGS`.
4242
* **Class-conditional:** add `--class_cond True` to `MODEL_FLAGS`.
4343

4444
Once you have setup your hyper-parameters, you can run an experiment like so:

0 commit comments

Comments
 (0)