Fixups for Continuous Normalizing Flow Tutorial

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.4k stars 130 forks source link

Closed atong01 closed 2 years ago

atong01 commented 2 years ago

Addresses #128.

Due to interface changes the CNF notebook did not work.

Changes:

Remove breaking lines looking for REQUIRESNOISE global variable in CNF redefinition.
Change s_span to t_span according to new interface
Reduce the number of epochs from 300 to 32 as this still gives reasonable results and is much faster on my cpu-only build.
Change indexing around as some of this has changed in the way NeuralODE models return data