search

DiffEqML / torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
https://torchdyn.org
Apache License 2.0
1.35k stars 125 forks source link

CNF interface changes #128

Closed atong01 closed 2 years ago

atong01 commented 2 years ago

Thank you for the great codebase. I've been looking into your implementation and tutorial of continuous normalizing flows. It seems that this has been broken for a while due to some interface changes in torchdyn. I managed to get it working with some small changes to indexing and naming.

If you're interested I could fix up and submit a PR.

joglekara commented 2 years ago

Thanks for using the library and providing your feedback!

Yes,torchdyn would love to have your input and bugfix so please feel free to submit a PR.

Zymrael commented 2 years ago

Thanks for #129, closing this for now!