Bert模型没有加载pickle

Question

我为NER训练了一个Bert模型。它工作得很好(显然需要时间来学习)。我将pickle保存为

with open('model_pkl', 'wb') as file:
pickle.dump(model, file)

当我试图加载这个保存的模型时，我得到以下错误。AttributeError: Can't get attribute 'BertModel' on <module '__main__' from '<input>'>。该方法适用于列表、字典等，但在pytorch模型上产生误差。我使用python 3.8.10

Answer 1

尝试使用火炬。Save and torch.load.