diff --git a/examples/cifar10/README.md b/examples/cifar10/README.md index 4201227..6fe081d 100644 --- a/examples/cifar10/README.md +++ b/examples/cifar10/README.md @@ -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.