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

Torch 2.0 compatible with Torchdyn #61

Closed kilianFatras closed 1 year ago

kilianFatras commented 1 year ago

What does this PR do?

This PR updates the code base compatible with Torch2.0 and TorchDyn1.0.6.

Fixes #60

Before submitting

Did you have fun?

Make sure you had fun coding 🙃

review-notebook-app[bot] commented 1 year ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

kilianFatras commented 1 year ago

Should we wait for torchdyn 1.0.7 or at least put a memo to discard these changes when 1.0.7 is out?