I am trying to train a model for token classification (NER), more or less following the example on the Huggingface course on Token Classification.
When I attempt to initiate arguments, using transformers.TrainingArguments, as below:
from transformers import TrainingArguments args = TrainingArguments( "bert-finetuned-ner", evaluation_strategy="epoch", save_strategy="epoch", learning_rate=2e-5, num_train_epochs=3, weight_decay=0.01, push_to_hub=True, )
I get the error:
RuntimeError: Failed to import transformers.training_args because of the following error (look up to see its traceback): /usr/local/lib/python3.7/dist-packages/_XLAC.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN2at13_foreach_erf_EN3c108ArrayRefINS_6TensorEEE
Why is this? Is this a bug for transformers on Colab GPUs?