qq_43711203的博客

原创 模型参数保存与使用

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