當(dāng)使用torch.load時(shí),報(bào)錯(cuò)No module named ‘models’
在網(wǎng)上查了很多資料說目錄結(jié)構(gòu)得和保存時(shí)一模一樣,話雖如此,但一直沒理解要如何一樣
最后調(diào)試發(fā)現(xiàn),No module named 'models’報(bào)錯(cuò)說沒有models,確實(shí)是因?yàn)闆]有.
比如下面:
我訓(xùn)練的用torch.save(model, checkpoint_path)保存的模型文件,那么會(huì)把對(duì)應(yīng)的網(wǎng)絡(luò)結(jié)構(gòu)路徑序列化到模型內(nèi)部, 而一旦更改了網(wǎng)絡(luò)結(jié)構(gòu)(model/slim)路徑,再torch.load()時(shí)會(huì)報(bào)錯(cuò).
解決方式一:
保持 網(wǎng)絡(luò)結(jié)構(gòu)(model/slim)路徑 不變.
解決方式二:torch
將調(diào)用 to’r’ch.load()方法的, xxx.py文件 與 模型的網(wǎng)絡(luò)結(jié)構(gòu)(slim)放在同一目錄下.文章來源:http://www.zghlxwxcb.cn/news/detail-627131.html
如圖:
在test.jw.py中調(diào)用torch.load()方法文章來源地址http://www.zghlxwxcb.cn/news/detail-627131.html
到了這里,關(guān)于ModuleNotFoundError: No module named ‘models‘解決torch.load問題【天坑】的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!