python torch报错 result = unpickler.load()ModuleNotFoundError: No module named 'models'
Posted
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了python torch报错 result = unpickler.load()ModuleNotFoundError: No module named 'models'相关的知识，希望对你有一定的参考价值。
参考技术A 将2个项目合并的时候，把文件夹A塞到文件夹B中，作为一个子文件夹，文件夹A中有一个已经训练好的.pth模型文件，B调用A以加载这个模型的时候，出现了题目中的报错，具体情况如下图时间久远，忘记这个.pth模型是怎么save的了，但是查了资料发现可能是因为这个模型保存了除模型参数之外其他的参数，因为涉及到文件的具体位置，所以在将A塞到B中的时候就会发生错误，由于旧的.pth保留的好，就将其重新load然后仅取其中的模型参数再save成新的.pth模型
···
package = torch.load(model_path)
p = './static_dict.pth'
torch.save(package.state_dict(), p)
···
···
model = Model_CNN()
model.load_state_dict(torch.load(model_path))
model.eval()
model = model.to(device)
···
以上是关于python torch报错 result = unpickler.load()ModuleNotFoundError: No module named 'models'的主要内容，如果未能解决你的问题，请参考以下文章
python pip install 报错TypeError: unsupported operand type(s) for -=: 'Retry' and 'int'