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

tqdm for the fixed odeint loop #153

Closed joglekara closed 10 months ago

joglekara commented 2 years ago

Probably need a way to suppress it or provide a flag for usage because the NN usecase is such that this would be unnecessary at best and cumbersome at worst

Zymrael commented 2 years ago

I agree that this is unnecessary (and adds a small overhead). What would be a use case for this? Logging step times and other metrics via a proper Solution class should cover most use cases I can think of.

joglekara commented 2 years ago

A use case might be a long time-scale PDE solve, ~10^5 steps, 0.2-0.5s per step.

Zymrael commented 2 years ago

this calls for a verbose option perhaps.