fast-bert
fast-bert copied to clipboard
Index Error
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