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

better handling for save_at types #157

Closed joglekara closed 2 years ago

joglekara commented 2 years ago

addresses a bug when save_at already comes in as a tensor

Zymrael commented 2 years ago

Let's keep these checks at the highest level (under odeint)