Skip to content

Commit

Permalink
update readme in cifar10 folder
Browse files Browse the repository at this point in the history
  • Loading branch information
kilianFatras committed Nov 4, 2023
1 parent bad11c1 commit 4b1e98a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/cifar10/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ To reproduce the experiments and save the weights, install the requirements from
python3 train_cifar10.py --model "otcfm" --lr 2e-4 --ema_decay 0.9999 --batch_size 128 --total_steps 400001 --save_step 20000
```

- For the Conditional Flow Matching method:
- For the Independant-Conditional Flow Matching (I-CFM) method:

```bash
python3 train_cifar10.py --model "icfm" --lr 2e-4 --ema_decay 0.9999 --batch_size 128 --total_steps 400001 --save_step 20000
Expand Down

0 comments on commit 4b1e98a

Please sign in to comment.