本文介绍了如何在Python中使用torch.save方法保存预训练的VGG16模型，包括结构和参数，以及推荐的state_dict方式。同时提到在导入模型时的注意事项，特别感谢B站up主tu堆的贡献。
本篇文章给大家谈谈python训练好的模型保存py后调用，以及python训练模型后怎么投入应用，希望对各位有所帮助，不要忘了收藏本站喔。
方式一：
#保存
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堆！！
扫一扫
1691
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
