modyfikacja trenera

2025-02-25 12:32:03 +01:00 · 2025-02-25 12:32:03 +01:00 · a380f06555
parent 1c348d41c0
commit a380f06555
1 changed files with 1 additions and 1 deletions
--- a/hft.py
+++ b/hft.py
@ -130,7 +130,7 @@ training_args = TrainingArguments(
 trainer = CustomTrainer(
    model=model,
    args=training_args,
-    train_dataset=tokenized_dataset["train"],
+    train_dataset=tokenized_dataset,
 )

 # Trening modelu