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

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐