Error Runtimeerror: expected all tensors to be on the same device
#2
by
olivierchen1015
- opened
Hi community,
does anyone know how to fix this error?
runtimeerror: 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__index_select)
(It seems this only happening on some models, but other models very well without any errors..) Thanks so much if anyone can help. :-)