torch.save(model.state_dict()
时间: 2023-04-26 19:06:22 浏览: 322
pytorch 状态字典:state_dict使用详解
5星 · 资源好评率100%
torch.save() 函数是用来保存 PyTorch 模型的状态字典 (state_dict) 的。用法为:torch.save(model.state_dict(), 'file_path.pt'),其中 'file_path.pt' 是保存的文件路径。
阅读全文
pytorch 状态字典:state_dict使用详解