a problem when train

[pokameng]

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

[DevTech15]

Are you using multiple GPUs for training?

[pokameng]

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.

[DevTech15]

Please try it on single GPU for now. Then we can try to train it on multiple GPUs.