我在这里不是在谈论一些著名的预训练模型(例如VGG,ResNet等),而是我训练的模型并将其保存为.pkl
文件。
例如,我训练了一个称为ANet的模型,并使用torch.save()
保存了该模型。
当我在新环境中编写模型测试代码并使用model = torch.load()
调用而未在文件中包含原始ANet类时,出现此错误返回:Attribute Error: Can't get attribute 'ANet' on <module '__main__'>
。
此错误消息是什么意思?如果我以前已经保存了模型并只想调用它,为什么还要包含整个类?