在Pytorch中使用预训练的模型时,我是否必须包含整个模型类?

时间:2019-01-03 02:20:19

标签: python deep-learning pytorch

我在这里不是在谈论一些著名的预训练模型(例如VGG,ResNet等),而是我训练的模型并将其保存为.pkl文件。

例如,我训练了一个称为ANet的模型,并使用torch.save()保存了该模型。

当我在新环境中编写模型测试代码并使用model = torch.load()调用而未在文件中包含原始ANet类时,出现此错误返回:Attribute Error: Can't get attribute 'ANet' on <module '__main__'>

此错误消息是什么意思?如果我以前已经保存了模型并只想调用它,为什么还要包含整个类?

0 个答案:

没有答案