A typo with use_gpu ? - Githubissues

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

A typo with use_gpu ? #66

Closed yanyuechuixue closed 1 year ago

yanyuechuixue commented 1 year ago

https://github.com/Photon-AI-Research/NeuralSolvers/blob/ef7c998d102c7b0e32bfdde73d44a60b68555bd6/PINNFramework/models/moe_finger.py#L170

StillerPatrick commented 1 year ago

You are Right. Can you Fix and test it. Looking forward to your PR 👌

yanyuechuixue commented 1 year ago

Sorry but I am a very beginner in this field. So it might take a long time for me..

StillerPatrick commented 1 year ago

@Jeyhun1 maybe you can help here as the maintainer. I think you just remove the .cuda line for non gpu execution

Jeyhun1 commented 1 year ago

Sure. I will fix and test it in a few days