Блог

Я попытался сохранить модель allennlp с помощью :

 with open("D:/allennlp2/modelallen.th", 'wb') as f:
    torch.save(model.state_dict(), f)

vocab.save_to_files("D:/allennlp2/vocabulary")
 

а потом я попытался загрузить его :

 vocab2 = Vocabulary.from_files("D:/allennlp2/vocabulary")

model2 = build_model(vocab2)
with open("D:/allennlp2/modelallen.th", 'rb') as f:
    models = model2.load_state_dict(torch.load(f))
 

У build_model есть аргумент для передачи vocab.
Он не загрузил модель, вместо этого выдал мне вывод :

Кто-нибудь знает, как загрузить модель без конфигурации с помощью allennlp?