Token Classification
GLiNER
PyTorch
English
NER
GLiNER
information extraction
encoder
entity recognition
modernbert

Finetuning error

#3
by seanbenhur - opened

I am trying to fine-tune this model, I updated gliner and transformers with latest versions but when I fine-tune, I am getting the below error

Asking to truncate to max_length but no maximum length is provided and the model has no predefined maximum length. Default to no truncation.
Error executing job with overrides: []
Traceback (most recent call last):
  File "/home/jovyan/rdk-log-analysis-framework/ner_research/training/train.py", line 102, in main
    train_result = trainer.train()
                   ^^^^^^^^^^^^^^^
  File "/home/jovyan/rdk_env/lib/python3.11/site-packages/transformers/trainer.py", line 2171, in train
    return inner_training_loop(
           ^^^^^^^^^^^^^^^^^^^^
  File "/home/jovyan/rdk_env/lib/python3.11/site-packages/transformers/trainer.py", line 2531, in _inner_training_loop
    tr_loss_step = self.training_step(model, inputs, num_items_in_batch)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: Trainer.training_step() takes 3 positional arguments but 4 were given

Sign up or log in to comment