superAE icon indicating copy to clipboard operation
superAE copied to clipboard

KeyError: 'unexpected key "decoder_s2s.embedding.weight" in state_dict'

Open M5218 opened this issue 6 years ago • 0 comments

Traceback (most recent call last): File "predict.py", line 177, in model.load_state_dict(checkpoints['model']) File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 522, in load_state_dict .format(name)) KeyError: 'unexpected key "decoder_s2s.embedding.weight" in state_dict' 在运行predict加载模型时,出现这个错误,请问下是什么原因呢?

M5218 avatar Jan 16 '19 14:01 M5218