如何保存pytorch 训练好的网络模型
Pytorch要保存训练好的模型,但是报错:PermissionError: [Errno 13] Permission denied。通过在相对路径的最后加上保存模型的文件名解决问题
·
这是Pytorch学习之路第三篇
问题
Pytorch中要用到以下代码保存训练好的模型(方法来源:如何保存与调用 pytorch 训练好的网络模型(转载))
torch.save(model.state_dict(), './model')
但是报错:PermissionError: [Errno 13] Permission denied
解决方案
相对路径的最后应该是保存模型的文件名。
torch.save(model.state_dict(), './model/model.pth')
效果
成功保存至指定路径
方案来源
pytorch save PermissionError: [Errno 13] Permission denied(转载)
更多推荐
所有评论(0)