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

方式一：
#保存
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堆！！