utterworks / fast-bert

Super easy library for BERT based NLP models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Index Error

bhavaygg opened this issue · comments

Hello,

I followed the documentation and converted my data files accordingly but i get the following error

 mp = predictor.predict_batch(test_data)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/fast_bert/prediction.py"$
    return self.learner.predict_batch(texts)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/fast_bert/learner_cls.py$
outputs = self.model(**inputs)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/torch/nn/modules/module.$
    result = self.forward(*input, **kwargs)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/transformers/modeling_be$
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/torch/nn/modules/module.$
    result = self.forward(*input, **kwargs)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/transformers/modeling_be$
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/torch/nn/modules/module.$
    result = self.forward(*input, **kwargs)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/transformers/modeling_be$
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/torch/nn/modules/module.$
    result = self.forward(*input, **kwargs)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/torch/nn/modules/sparse.$
    self.norm_type, self.scale_grad_by_freq, self.sparse)
  File "/home/bhavay18384/.conda/envs/myenv/lib/python3.7/site-packages/torch/nn/functional.py",$
    return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
IndexError: index out of range in self

I got same error