同时用keras+tensorflow和pytorch真的是容易走火入魔,写着写着总能漏点啥,现在想加载训练好的参数:
model = my_model()
model.load_state_dict(the_path)
但是报错如下:
File "C:\Python\Python35\site-packages\torch\nn\modules\module.py", line 476, in load_state_dict
for name, param in state_dict.items():
AttributeError: 'str' object has no attribute 'items'
其实是因为load_state_dict的参数是torch.saved保存的对象,写的时候忘了写torch.load。正确代码如下
model = my_model()
model.load_state_dict(torch.load(the_path))