当使用torch.load时,报错No module named ‘models’
在网上查了很多资料说目录结构得和保存时一模一样,话虽如此,但一直没理解要如何一样
最后调试发现,No module named 'models’报错说没有models,确实是因为没有.
比如下面:
我训练的用torch.save(model, checkpoint_path)保存的模型文件,那么会把对应的网络结构路径序列化到模型内部, 而一旦更改了网络结构(model/slim)路径,再torch.load()时会报错.
解决方式一:
保持 网络结构(model/slim)路径 不变.
解决方式二:torch
将调用 to’r’ch.load()方法的, xxx.py文件 与 模型的网络结构(slim)放在同一目录下.文章来源:https://www.toymoban.com/news/detail-627131.html
如图:
在test.jw.py中调用torch.load()方法文章来源地址https://www.toymoban.com/news/detail-627131.html
到了这里,关于ModuleNotFoundError: No module named ‘models‘解决torch.load问题【天坑】的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!