test_sample.py运行报错，貌似是device不一样，untimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper_CUDA_gather)
WUHU-G opened this issue · comments
是否已经解决这个问题了呢，我也遇到了相同的问题 @FastBCSD @chaoyi-wu @xiaoman-zhang
我已经解决这个问题了，转移到一个就行了