a problem when train
pokameng opened this issue · comments
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
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.