New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

Qinger27 opened this issue Oct 18, 2023 · 1 comment

Qinger27 commented Oct 18, 2023

你好！
我在40G的A100卡上训练您的burgers方程的deepONet.py文件时，报错oom：

请问下怎么修改来完成多卡训练呀？抱歉我对DeepXDE的使用不是很熟悉，期待回复~

Member

lululxvi commented Nov 19, 2023

The demo code shouldn't use so much memory.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment