python/pytorch中如何保存和导入已经训练好的模型_导入torch模型

作者：码创造者 | 2024-07-13 07:42:47

踩

导入torch模型

方式一：
#保存
torch.save（vgg16，“vgg16_method1.pth”） #保存模型结构和模型里面的参数
#加载
model=torch.load（“vgg16_method1.pth”）

方式2：
torch.save（vgg16.state_dict(),“vgg16_method2.pth”）#保存模型参数（官方推荐）

vgg16 = torchvision.model.vgg16（pretrained=False）
vgg16.load_state_dict(torch.load("vgg16_method2.pth"))

方式1的话引入的时候程序要可以访问到模型定义的function （说白了就是在引入文件中import一下或者把代码class/def的内容再拷过去就可以）

感谢B站up主tu堆！！

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/码创造者/article/detail/818324