donglixp / coarse2fine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Fail to load the pretrained model

opened this issue · comments

I wonder whether the pretrain.pt is the pretrained model that can be used directly? I tried to run evaluate.py via loading it but got this error:

  File "/logical/table/Translator.py", line 112, in __init__
    checkpoint = torch.load(opt.model)
  File "/opt/conda/lib/python3.6/site-packages/torch/serialization.py", line 303, in load
    return _load(f, map_location, pickle_module)
  File "/opt/conda/lib/python3.6/site-packages/torch/serialization.py", line 469, in _load
    result = unpickler.load()
ModuleNotFoundError: No module named 'table'

Do you have any idea about it? Thanks a lot.