模型参数保存与使用

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