我是在用load.torch加载网络模型的.pt文件时遇到这个问题的。

虽然我的python代码中没有调用models文件夹中的任何东西,但是遇到了该问题

解决方法:现在的.py文件目录下,创建一个文件夹,将其命名为:models。

因为我的模型训练时用到了vgg.py,xxxx.py,yyyyy.py三个文件夹,并且我模型参数.pt文件中的数据也是与这三个py文件有关,所以我将这三个.py文件加到了models文件夹下,最后代码能够顺利运行。

希望能对你有所帮助

参考:​​​​​​ModuleNotFoundError: No module named ‘models‘解决torch.load问题【天坑】__less is more的博客-CSDN博客

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐