modyfikacja trenera

This commit is contained in:
l.gabrysiak 2025-02-25 12:32:03 +01:00
parent 1c348d41c0
commit a380f06555
1 changed files with 1 additions and 1 deletions

2
hft.py
View File

@ -130,7 +130,7 @@ training_args = TrainingArguments(
trainer = CustomTrainer( trainer = CustomTrainer(
model=model, model=model,
args=training_args, args=training_args,
train_dataset=tokenized_dataset["train"], train_dataset=tokenized_dataset,
) )
# Trening modelu # Trening modelu