search

Pay20Y / SEED

163 stars 39 forks source link

can not load checkpoints to evaluate model metrics #29

Closed tink2123 closed 3 years ago

tink2123 commented 3 years ago
  1. download checkpoints
  2. sh test.sh

but i got:

=> Loaded checkpoint '/workspace/tools/SEED/demo.pth.tar'
Traceback (most recent call last):
  File "main.py", line 268, in <module>
    main(args)
  File "main.py", line 184, in main
    model.load_state_dict(checkpoint['state_dict'])
  File "/usr/local/lib/python3.7/site-packages/torch/nn/modules/module.py", line 777, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for ModelBuilder:
        Missing key(s) in state_dict: "decoder.decoder.embed_fc.weight", "decoder.decoder.embed_fc.bias", "embeder.eEmbed.weight", "embeder.eEmbed.bias".