分析报错原因

解决办法

train.py文件中,找到一下两行代码,注释掉

以防后续运行代码出错,训练结束后,记得把这两行代码打开

在这里插入图片描述

还有个方法(我在调用test.py也遇到了找不到这个pt文件的错误):

直接将pt文件拷贝到代码根目录

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐