search

shunsukesaito / SCANimate

This repository contains the code for the paper "SCANimate: Weakly Supervised Learning of Skinned Clothed Avatar Networks"
https://scanimate.is.tue.mpg.de/
Other
273 stars 31 forks source link

Tensors are on different gpu when training with "cuda:1" #8

Closed coreqode closed 2 years ago

coreqode commented 2 years ago

https://github.com/shunsukesaito/SCANimate/blob/f2eeb5799fd20fd9d5933472f6aedf1560296cbe/apps/train_scanimate.py#L223

Please change .cuda() to .to(cuda). It gives error, when the cuda is not cuda:0