@@ -282,7 +282,7 @@ At this point, only three steps remain:
|
||||
... args=training_args,
|
||||
... train_dataset=encoded_minds["train"],
|
||||
... eval_dataset=encoded_minds["test"],
|
||||
... tokenizer=processor.feature_extractor,
|
||||
... tokenizer=processor,
|
||||
... data_collator=data_collator,
|
||||
... compute_metrics=compute_metrics,
|
||||
... )
|
||||
|
||||
@@ -469,7 +469,7 @@ Instantiate the `Trainer` object and pass the model, dataset, and data collator
|
||||
... train_dataset=dataset["train"],
|
||||
... eval_dataset=dataset["test"],
|
||||
... data_collator=data_collator,
|
||||
... tokenizer=processor.tokenizer,
|
||||
... tokenizer=processor,
|
||||
... )
|
||||
```
|
||||
|
||||
|
||||
Reference in New Issue
Block a user