search

lu-group / deeponet-fno

DeepONet & FNO (with practical extensions)
Other
232 stars 49 forks source link

请问deepONet_POD如何设置多卡训练 #13

Open Qinger27 opened 1 year ago

Qinger27 commented 1 year ago

你好！
我在40G的A100卡上训练您的burgers方程的deepONet.py文件时，报错oom： image 请问下怎么修改来完成多卡训练呀？抱歉我对DeepXDE的使用不是很熟悉，期待回复~

lululxvi commented 1 year ago

The demo code shouldn't use so much memory.