-
Notifications
You must be signed in to change notification settings - Fork 48
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.
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
a problem when train #21
Comments
Are you using multiple GPUs for training? |
Yeah 4 gpus
…---Original---
From: ***@***.***>
Date: Tue, May 11, 2021 07:29 AM
To: ***@***.***>;
Cc: ***@***.******@***.***>;
Subject: Re: [NVlabs/STEP] a problem when train (#21)
Are you using multiple GPUs for training?
—
You are receiving this because you authored the thread.
Reply to this email directly, view it on GitHub, or unsubscribe.
|
Please try it on single GPU for now. Then we can try to train it on multiple GPUs. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
RuntimeError: A tensor was not on the same device as the first tensor
When i run the train.py ,i face up with this problem
The text was updated successfully, but these errors were encountered: