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