爱吃肉的鹏 2022-05-05 13:48 采纳率: 25%
浏览 45
已结题

有关pytorch模型保存和加载问题

pytorch环境1.7
以前的时候一直使用torch.save(model.state_dict(), 'mymodel.pth')来保存模型,不过这种保存形式只是保存了模型的权重而言,同时在模型加载的时候需要对模型实例化后才能加载权重。

于是我便利用torch.save(model, 'mymodel.pth')将模型的图结构和权重一起保存,这样在模型加载的时候就可以直接加载了,但发现了一个问题:
当我用这种方式保存pth文件后,在当前路径下可以正常加载,但如果我将这个模型文件pth拿到其他地方去,就会报错,比如报以下错误:

 model = torch.load('whole_model.pth')

ModuleNotFoundError: No module named 'nets‘

报错中的nets是我以前存放定义的model的类文件路径。那是不是这种保存方式是将原来工程中的nets下所有相关的属性都保存了??那如果遇到这种问题应该怎么做呢?因为我是想训练完以后的模型,未来可以在其他地方进行一个模型的部署的。

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 5月13日
    • 创建了问题 5月5日

    悬赏问题

    • ¥20 Html备忘录页面制作
    • ¥15 黄永刚的晶体塑性子程序中输入的材料参数里的晶体取向参数是什么形式的?
    • ¥20 数学建模来解决我这个问题
    • ¥15 计算机网络ip分片偏移量计算头部是-20还是-40呀
    • ¥15 stc15f2k60s2单片机关于流水灯,时钟,定时器,矩阵键盘等方面的综合问题
    • ¥15 YOLOv8已有一个初步的检测模型,想利用这个模型对新的图片进行自动标注,生成labellmg可以识别的数据,再手动修改。如何操作?
    • ¥30 NIRfast软件使用指导
    • ¥20 matlab仿真问题,求功率谱密度
    • ¥15 求micropython modbus-RTU 从机的代码或库?
    • ¥15 django5安装失败