modyfikacja trenera
This commit is contained in:
parent
1c348d41c0
commit
a380f06555
2
hft.py
2
hft.py
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue