search

utterworks / fast-bert

Super easy library for BERT based NLP models
Apache License 2.0
1.86k stars 341 forks source link

Issue with BertNERPredictor.predict_batch(array) #298

Open AshayT opened 3 years ago

AshayT commented 3 years ago

After reloading the saved trained NER model, facing an issue with BertNERPredictor.predict_batch(array). This gives- IndexError: index out of range in self

Same thing on google colab (GPU) gives- RuntimeError: CUDA error: device-side assert triggered

BertNERPredictor.predict(text) works perfectly fine!