You need the second input, the description of your network: https://neuralpde.sciml.ai/dev/examples/ode/
1 Like
You need the second input, the description of your network: https://neuralpde.sciml.ai/dev/examples/ode/