weixin_44263065的博客

原创 解决pytorch加载模型报错TypeError: ‘collections.OrderedDict‘ object is not callable

之前训练完一个模型，模型参数用代码torch.Save(model.state_dict(), 'data/model.pth')保存道model.pth文件中，用load语句加载文件时报错，出现上面的报错语句。看了一下文献，发现原因是model.pth文件中保存的是模型的参数，不能直接加载，需要先导入网络结构，再将文件中的网络结构的参数导入。