search

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

Horovod Support #15

Closed StillerPatrick closed 3 years ago

StillerPatrick commented 3 years ago

Horovod support is now enabled you can add the use_horovod function in the fit function of the PINN. Pls check the code if everything is fine.