mod
This commit is contained in:
parent
90f3d66723
commit
f3e84a6fc0
1
hft.py
1
hft.py
|
|
@ -178,7 +178,6 @@ training_args = TrainingArguments(
|
||||||
save_steps=1000,
|
save_steps=1000,
|
||||||
report_to="none",
|
report_to="none",
|
||||||
gradient_checkpointing=True,
|
gradient_checkpointing=True,
|
||||||
evaluation_strategy="epoch", # częstotliwość oceny
|
|
||||||
per_device_train_batch_size=8, # batch size dla treningu
|
per_device_train_batch_size=8, # batch size dla treningu
|
||||||
per_device_eval_batch_size=8, # batch size dla ewaluacji
|
per_device_eval_batch_size=8, # batch size dla ewaluacji
|
||||||
logging_dir='./logs' # folder do logów
|
logging_dir='./logs' # folder do logów
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue