Cannot load model via torch.load

[MariamRiaz]

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]

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