yolact RuntimeError: Attempted to set the storage of a tensor on device "cuda:0" to a storage on different device "cpu". This is no longer allowed; the devices must match.

RuntimeError: Attempted to set the storage of a tensor on device "cuda:0" to a storage on different device "cpu". This is no longer allowed; the devices must match.

Open Urwah92 opened this issue 2 years ago • 3 comments

If I try to train custom data, I got this error. Please, if anyone can help.

Thanks.

Jul 31 '22 12:07 Urwah92

See:

https://github.com/dbolya/yolact/issues/755#issuecomment-1141917563

Solved same issue for me.

Oct 28 '22 10:10 cdrwolfe

我也的到了这个错误，但是还没找到办法

May 13 '23 23:05 LLF-afk

我已经找到了解决办法，https://blog.csdn.net/zhang2h/article/details/125941648，参考这个博客，更换使用的预训练model权重

May 14 '23 00:05 LLF-afk