Photon-AI-Research / NeuralSolvers

Neural network based solvers for partial differential equations and inverse problems :milky_way:. Implementation of physics-informed neural networks in pytorch.
MIT License
141 stars 46 forks source link

Fix correct checkpointing #34

Closed StillerPatrick closed 3 years ago

StillerPatrick commented 3 years ago

Update in checkpointing

In the current setup every n time-step a checkpoint was written, without overwriting the previous one. Which costs a lot of memory. However, the old checkpoints are in the most cases not considered anymore.

Update of the 1D schrodinger

The 1D schrodinger example was outdated and will be updated with this pull request.