feat: modified fit() function in aug_ode

DynaDojo / dynadojo

An Extensible Benchmarking Platform for Scalable Dynamical System Identification

https://dynadojo.github.io/dynadojo/

MIT License

14 stars 11 forks source link

Closed sameerashahh closed 2 months ago

sameerashahh commented 2 months ago

Modified the fit function to return training loss iteratively, after each epoch.