- 博客(1)
- 收藏
- 关注
原创 模型参数保存与使用
(1)训练结束保存最优化模型参数:filepath = os.path.join(save_path, 'check.pth'.format(model_name))#save_path:../自定义(自己设置放置位置)torch.save(net.state_dict(), filepath)#net:模型(2)验证加载使用:et.load_state_dict(torch.load("../checkpoint/check.pth"))#load内容为保存时位置文件名
2022-03-02 20:24:15
261
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人