Skip to content

Commit

Permalink
Update readme with links to saved weights
Browse files Browse the repository at this point in the history
  • Loading branch information
atong01 committed Nov 1, 2023
1 parent bc91ce1 commit 62b27d1
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion examples/cifar10/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,11 @@ To compute the FID from the OT-CFM model at end of training, run:
python3 compute_fid.py --model "otcfm" --step 400000 --integration_method dopri5
```

For the other models, change the "otcfm" argument by "cfm" or "fm". For easy reproducibility of our results, we will provide soon the saved weights of our different methods trained for 400000 iterations.
For the other models, change the "otcfm" argument by "cfm" or "fm". For easy reproducibility of our results, you can download the model weights at 400000 iterations here:

[cfm weights](https://github.com/atong01/conditional-flow-matching/releases/download/1.0.4/cfm_cifar10_weights_step_400000.pt)
[otcfm weights](https://github.com/atong01/conditional-flow-matching/releases/download/1.0.4/otcfm_cifar10_weights_step_400000.pt)
[fm weights](https://github.com/atong01/conditional-flow-matching/releases/download/1.0.4/fm_cifar10_weights_step_400000.pt)

To recompute the FID, change the PATH variable with where you have saved the downloaded weights.

Expand Down

0 comments on commit 62b27d1

Please sign in to comment.