fix bug related to device id in `conditional_mnist.ipynb`

atong01 / conditional-flow-matching

TorchCFM: a Conditional Flow Matching library

https://arxiv.org/abs/2302.00482

MIT License

1.25k stars 101 forks source link

Closed ImahnShekhzadeh closed 10 months ago

ImahnShekhzadeh commented 10 months ago

This PR fixes a bug related to the device ID in a tutorial.

[y] Did you make sure title is self-explanatory and the description concisely explains the PR?
[y] Did you make sure your PR does only one thing, instead of bundling different changes together?
[ ] Did you list all the breaking changes introduced by this pull request?
[ ] Did you test your PR locally with pytest command?
[ ] Did you run pre-commit hooks with pre-commit run -a command?

review-notebook-app[bot] commented 10 months ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

atong01 commented 10 months ago

Thanks for this fix!