0
点赞
收藏
分享

微信扫一扫

pytorch-模型保存

四月Ren间 2022-03-15 阅读 53

pytorch框架下两种模型保存及加载方式

方法1

以字典(key-value)保存模型的参数

torch.save(vgg16.state_dict(), "路径")
vgg16.load_state_dict(torch.load("路径"))

方法2

保存模型参数及模型结构

torch.save(vgg16, "路径h")
model = torch.load("路径")
举报

相关推荐

0 条评论