pytorch框架下两种模型保存及加载方式
方法1
以字典（key-value）保存模型的参数
torch.save(vgg16.state_dict(), "路径")
vgg16.load_state_dict(torch.load("路径"))
方法2
保存模型参数及模型结构
torch.save(vgg16, "路径h")
model = torch.load("路径")
pytorch-模型保存
阅读 57
2022-03-15
相关推荐
精彩评论（0）
0 0 举报
pytorch-模型保存
阅读 57
2022-03-15
pytorch框架下两种模型保存及加载方式
方法1
以字典（key-value）保存模型的参数
torch.save(vgg16.state_dict(), "路径")
vgg16.load_state_dict(torch.load("路径"))
方法2
保存模型参数及模型结构
torch.save(vgg16, "路径h")
model = torch.load("路径")
相关推荐
精彩评论（0）