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

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

暂无评论

发表评论

相关推荐

SSD网络及代码理解

写在前面:在读了SSD的论文之后,完整的看了一遍SSD的代码,有了许多体会,以此记录自己的学习过程。 论文传送门:SSD: Single Shot MultiBox Dete

pytorch建立mobilenetV3-ssd网络并进行训练与预测

前言 这篇文章记录的是我在公司实习用深度学习做车辆信息识别项目时,用来做车辆检测的算法。因为我们公司面向的边缘端计算,边缘盒子的计算能力有限,所以我们在做算法研究时,就尽量选用轻量级算法