diff --git a/examples/cifar10/README.md b/examples/cifar10/README.md index 6fe081d..70319f8 100644 --- a/examples/cifar10/README.md +++ b/examples/cifar10/README.md @@ -34,9 +34,11 @@ 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, 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) +- [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.