search

DEEPDIP-project / CoupledNODE.jl

Apache License 2.0
2 stars 0 forks source link

Make `create_f_CNODE` also return the NODE #36

Open SCiarella opened 1 month ago

SCiarella commented 1 month ago

In this way there is no need for the user to call directly NeuralODE(*).

Then, create_f_CNODE will need to receive as input also the parameters that will be used for the solution, like t_range, algorithm, ecc..

luisaforozco commented 1 month ago

If the force is needed, for example to create the loss, one could get it via: my_neuralODE.model

luisaforozco commented 1 month ago

This is implemented now in branch wrap_neuralODE.

However, I see that there are some sort of memory leaks (many allocations) that were not there before. We need to first understand what is going on before merging that work.