parent
91f4c84a19
commit
d3cbc997a2
|
@ -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,
|
||||
... )
|
||||
```
|
||||
|
||||
|
|
Loading…
Reference in New Issue