Fan Lin

icon company Alibaba icon location Hangzhou, China

Results 91 comments of Fan Lin

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:7 and cuda:0! (when checking argument for argument index in method wrapper_CUDA__index_select)

try 0.2.8 version, and remove the device_map="balanced" parameter, it works for me.