RuntimeError: storage has wrong size: expected XXX got XXX 以及多卡GPU训练转到多卡GPU测试

最新推荐文章于 2024-07-24 20:25:39 发布

程序员毛师傅

最新推荐文章于 2024-07-24 20:25:39 发布

阅读量3.4k

点赞数

分类专栏：深度学习文章标签：算法计算机视觉机器学习

本文链接：https://blog.csdn.net/qq_42035274/article/details/124193207

版权

在训练时，我的代码出现了以下问题：

Traceback (most recent call last):
  File "/root/PycharmProjects/test.py", line 8, in <module>
    model_dict = torch.load("0.pth")
  File "torch/serialization.py", line 529,

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

程序员毛师傅

关注关注

0
点赞
踩
2

收藏

觉得还不错? 一键收藏
打赏
0
评论
RuntimeError: storage has wrong size: expected XXX got XXX 以及多卡GPU训练转到多卡GPU测试

在训练时，我的代码出现了以下问题：Traceback (most recent call last): File "/root/PycharmProjects/test.py", line 8, in <module> model_dict = torch.load("0.pth") File "torch/serialization.py", line 529, in load return _legacy_load(opened_file, map_locati.
复制链接

扫一扫