(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内容为保存时位置文件名