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.36k stars 125 forks source link

Outputing sequences #68

Closed drozzy closed 3 years ago

drozzy commented 3 years ago

I'm looking to taking in a sequence of values and output a sequence as well. So instead of doing flatten and computing the final output from neural ode, I would like to make use of each intermediate value that neural ode outputs and produce a loss with a target label.

Is there an example of this somewhere using TorchDyn?

Thanks

Zymrael commented 3 years ago

This is the default behavior now with NeuralODE instances. The solution will be a Tensor of output num_time_points, batch, dims, which you can calculate losses over.