# 模型的存储
torch.save(xxx_model.state_dict(),'./model_param.pth')
model = Net().to(device)
model.load_state_dict(torch.load('./model_param.pth'))
model参数的存储和调用
最新推荐文章于 2022-11-13 13:14:04 发布
# 模型的存储
torch.save(xxx_model.state_dict(),'./model_param.pth')
model = Net().to(device)
model.load_state_dict(torch.load('./model_param.pth'))
2590

被折叠的 条评论
为什么被折叠?