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

Provide a seperate pretraining function #60

Open StillerPatrick opened 2 years ago

StillerPatrick commented 2 years ago

At the moment, you need to set the number of main training epochs to 0, activate the pretaining function, and deactivate lbfgs fine-tuning. Here it would be better to have a separate function that trains initial condition only. This helps to find a suitable architecture.