Bugfix/higher order - Githubissues

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

Closed joglekara closed 2 years ago

joglekara commented 2 years ago

fixes bugs in

higher-order forward layer. There was a bug in the concatenate call
notebooks - related to the fact that calling a NODE now returns a t_span and trajectory. This requires writing out the operation within the training step rather than using a nn.Sequential to build the training step.