Chapter 3 questions

Course
149

Hi guys, Can anyone help with this ? Getting this error when running the code in colab notebook .

TypeError                                 Traceback (most recent call last)
<ipython-input-20-33ccf89368f8> in <cell line: 0>()
      1 from transformers import TrainingArguments, Trainer
----> 2 training_args = TrainingArguments("test-trainer", evaluation_strategy="epoch")
      3 model = AutoModelForSequenceClassification.from_pretrained(checkpoint, num_labels=2)
      4 
      5 trainer = Trainer(

TypeError: TrainingArguments.__init__() got an unexpected keyword argument 'evaluation_strategy'

For the second line of this code

training_args = TrainingArguments("test-trainer", eval_strategy="epoch")
model = AutoModelForSequenceClassification.from_pretrained(checkpoint, num_labels=2)

trainer = Trainer(
    model,
    training_args,
    train_dataset=tokenized_datasets["train"],
    eval_dataset=tokenized_datasets["validation"],
    data_collator=data_collator,
    tokenizer=tokenizer,
    compute_metrics=compute_metrics,
)
1 Like