Releases: Photon-AI-Research/NeuralSolvers
Geometries and Adaptive Sampling
We added Geometries as a new interface for the PDE dataset. This
allows a generic inference pipeline and reduces the amount of code to
solve a PDE with NeuralSolvers. This includes various sampling methods and brings adaptive sampling into Neural Solvers.
Furthermore, we improved the prints and the logging features.
Tensorboard-logging, gradient visualisation, bugfixes
New Features
Tensorboard logging is available and can be activated in the fit() function. Logger enables automatic tracking of all loss terms and their weights
Logging of Loss gradients in order to identity gradient pathologies, you can track the gradient updates by adding a logger and setting the
track_gradient flag in the fit() function.
Bugfixes
Fix the usage of the to method in MLP and FingerNet
Examples
A working example of the heat equation
Learning Rate Annealing, Logger, Callbacks and Bugfixes
New Features
Learning Rate Annealing is now available and can be activated in the fit()function. This algorithm balances the weights for initial and boundary conditions depending on the stiffness of the underlying PDE
Logger are available and can be activated in the fit()function. Logger enable automatic tracking of all loss terms and theire weights for example with Weights and Biases which is implemented yet. By implementing the Logger_Interface you can build your own loggers.
Pretraining the fit() function allows a pretraining by optimizing only on the initial condition
Callbacks its possible to write custom callbacks in order to modify the behavior in the training loop. At the moment only callbacks at the end of epochs are supported
Bugfixes
Fix the bug in the calculation of neumann and robin boundary condition
Examples
A working example of burgers equation