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!
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!