raise RuntimeError(‘Attempting to deserialize object on a CUDA ‘ RuntimeError: Attempting to deseri

 raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

rel_extractor.load_state_dict(torch.load(model_state_path))
改成
rel_extractor.load_state_dict(torch.load(model_state_path, map_location=‘cpu’))

版权声明:本文为CSDN博主「愚昧之山绝望之谷开悟之坡」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_15821487/article/details/122554992

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

用mmdetection跑通Mask-RCNN

数据准备工作
原始数据分两个文件夹,images(彩色图片)、mask(掩膜照片,一定要转换成灰度图images的名字若为“ABC.jpg”,mask里的掩膜

【学习分享】目标检测中的锚框(Anchor)

锚框
引例
在理解目标检测的锚框之前,我们首先通过一个不太严谨的例子对锚框进行一个简单的了解: 由于目前污染比较严重,导致海洋中漂浮着许多垃圾,这些垃圾既污染环境,又不利