Я пытаюсь загрузить вес модели AllenNLP. Я не смог найти никакой документации о том, как сохранить / загрузить всю модель, поэтому играл только с весами.
from allennlp.nn import util
model_state = torch.load(filename_model, map_location=util.device_mapping(-1))
model.load_state_dict(model_state)
Я немного изменил свой входной корпус, и я предполагаю, что из-за этого я получаю несоответствие размеров корпуса:
RuntimeError: Error(s) in loading state_dict for BasicTextFieldEmbedder:
size mismatch for token_embedder_tokens.weight:
copying a param with shape torch.Size([2117, 16]) from checkpoint,
the shape in current model is torch.Size([2129, 16]).
Кажется, нет официального способа сохранить модель с помощью словаря корпусов. Есть ли какие-нибудь хаки?