NeuroDiffGym / neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.
http://pypi.org/project/neurodiffeq/
MIT License
702 stars 90 forks source link

custom loss function #29

Closed feiyu-chen96 closed 4 years ago

feiyu-chen96 commented 4 years ago

Currently, the loss function is assumed to be a function of the reparameterized output of the neural net alone. What if the user want a loss function that also takes the independent variable directly as an input?