Cannot load model via torch.load

nlpyang / PreSumm

code for EMNLP 2019 paper Text Summarization with Pretrained Encoders

MIT License

1.29k stars 465 forks source link

Open MariamRiaz opened 2 years ago

MariamRiaz commented 2 years ago

I am trying to load a pretrained model in my notebook. Using google colab for the experiment,

import torch

PATH = "path/to/the/model/model_step_148000.pt" model = torch.load(PATH)

it gives me this error ; ModuleNotFoundError: No module named 'models'

How can I solve this?

SabrinaZhuangxx commented 2 years ago

Do you know how to continue training the ext model？thank you