saving a model leads to an error

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

saving a model leads to an error #21

Closed maxxxzdn closed 3 years ago

maxxxzdn commented 3 years ago

https://github.com/ComputationalRadiationPhysics/NeuralSolvers/blob/2a9457257b35265d219db245eb765c6132559748/PINNFramework/PINN.py#L126

Seems like it tries to get the attribute "hpm_model" of the method "pinn_loss" which leads to an error.

StillerPatrick commented 3 years ago

No, the pinn loss is a object that should contain a hpm model inside.