RuntimeError: Attempted to set the storage of a tensor on device “cuda:0“ to a storage on different

在运行yolact++模型时,出现如题的错误。


解决办法为找到报错点209行


在yolact.py的位置添加--------------,map_location='cuda:0'

 

RuntimeError: Attempted to set the storage of a tensor on device “cuda:0“ to a storage on different_第1张图片

你可能感兴趣的:(pytorch,深度学习,python)