chuanmeng / dukenet Goto Github PK

CUDA_version:10.1
cudnn_version:7603
Vocabulary size 30522
The number of dev_samples: 8814
The number of test_seen_samples: 4336
The number of test_unseen_samples: 4370
Start inference at epoch d_3
Traceback (most recent call last):
  File "DukeNet/Run.py", line 321, in <module>
    inference(args)
  File "DukeNet/Run.py", line 261, in inference
    inference(args.dataset, i)
  File "DukeNet/Run.py", line 206, in inference
    model.load_state_dict(torch.load(file)["model"])
  File "/home/bakhtiyar/anaconda3/envs/dukenet-2/lib/python3.6/site-packages/torch/nn/modules/module.py", line 830, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DukeNet:
	Missing key(s) in state_dict: "enc.enc.embeddings.position_ids", "dec.embedding.0.position_ids", "dec.embedding.0.word_embeddings.weight", "dec.embedding.0.position_embeddings.weight", "dec.embedding.0.token_type_embeddings.weight", "dec.embedding.0.LayerNorm.weight", "dec.embedding.0.LayerNorm.bias", "dec.embedding.1.pe". 
	Unexpected key(s) in state_dict: "dec.embedding.word_embeddings.weight", "dec.embedding.position_embeddings.weight", "dec.embedding.token_type_embeddings.weight", "dec.embedding.LayerNorm.weight", "dec.embedding.LayerNorm.bias".