Я пытаюсь обучить модель pytorch на colab, затем сохранить параметры модели и загрузить ее на свой локальный компьютер.
После тренировки параметры модели сохраняются, как показано ниже:
torch.save(Model.state_dict(),PATH)
загружено как показано ниже:
device = torch.device('cpu')
Model.load_state_dict(torch.load(PATH, map_location=device))
ошибка:
AttributeError: 'Sequential' object has no attribute 'copy'
Кто-нибудь знает, как решить эту проблему?