bugfixes for args in steps

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

You have to pass it everywhere that you pass (t, x)

It would be good to just take **kwargs like pytorch does

Zymrael commented 2 years ago

LGTM