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
664 stars 87 forks source link

APTx Function: Default value of gamma should be 0.5 when trainable=False #222

Closed mr-ravin closed 5 days ago

mr-ravin commented 2 weeks ago

In APTx activation function, the default value of gamma should be 0.5 when trainable=False. One can verify it in the APTx paper here: https://arxiv.org/pdf/2209.06119

shuheng-liu commented 5 days ago

Thank you for this PR