这是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(转载)