XLNetForSequenceClassification Предварительно обученная модель не может быть загружена - PullRequest
Новая
       76

XLNetForSequenceClassification Предварительно обученная модель не может быть загружена

0 голосов
/ 25 апреля 2020

Я пытался загрузить X LNet с предварительной подготовкой, но это произошло. Я пробовал это раньше, и это сработало, однако, теперь это не так. Любое предложение о том, как решить эту проблему? 

model = XLNetForSequenceClassification.from_pretrained("xlnet-large-cased", num_labels = 2)
model.to(device)

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-55-d6f698a3714b> in <module>()
----> 1 model = XLNetForSequenceClassification.from_pretrained("xlnet-large-cased", num_labels = 2)
      2 model.to(device)

3 frames
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/sparse.py in __init__(self, num_embeddings, embedding_dim, padding_idx, max_norm, norm_type, scale_grad_by_freq, sparse, _weight)
     95         self.scale_grad_by_freq = scale_grad_by_freq
     96         if _weight is None:
---> 97             self.weight = Parameter(torch.Tensor(num_embeddings, embedding_dim))
     98             self.reset_parameters()
     99         else:

RuntimeError: Trying to create tensor with negative dimension -1: [-1, 1024]

1 Ответ

0 голосов
/ 26 апреля 2020

Если вы ничего внутри не изменили, скорее всего, несовпадение версий. Вы обновили какие-либо соответствующие модули? Go вернуться к предыдущей версии, если она у вас есть, которая должна ее решить.

Квантование Pytorch RuntimeError: Попытка создать тензор с отрицательным измерением

...