Port the 2D Schrodinger to the framework

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

Port the 2D Schrodinger to the framework #18

Open StillerPatrick opened 3 years ago

StillerPatrick commented 3 years ago

[ ] Split the EC Dataset into three datasets
[ ] Implement the Normalization Condition as a new designed Boundary Condition (https://pytorch.org/docs/stable/generated/torch.trapz.html) could make things easier
[ ] Integrate the new normalization condition into the PINN loss calculation
[ ] Switch from x,y,t representation to a single tensor that represents all cases
[ ] Integrate the implementation of the analytical solution in the IC dataset

StillerPatrick commented 3 years ago

[ ] Add pretraining on intital condition