M5218

Hi， Thanks your code very much, How do I get the number of words generated by the launch model in one state? For example, in the paper of Figure 3,...

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加载模型时，出现这个错误，请问下是什么原因呢？

ImportError: cannot import name 'masked_cross_entropy'

3

您好，请问能提供一下masked_cross_entropy该函数的代码吗？谢谢