mod
This commit is contained in:
parent
5c0e4381e7
commit
0c01db5995
4
hft.py
4
hft.py
|
|
@ -178,8 +178,8 @@ training_args = TrainingArguments(
|
||||||
save_steps=1000,
|
save_steps=1000,
|
||||||
report_to="none",
|
report_to="none",
|
||||||
gradient_checkpointing=True,
|
gradient_checkpointing=True,
|
||||||
per_device_train_batch_size=8, # batch size dla treningu
|
per_device_train_batch_size=4, # batch size dla treningu
|
||||||
per_device_eval_batch_size=8, # batch size dla ewaluacji
|
per_device_eval_batch_size=4, # batch size dla ewaluacji
|
||||||
logging_dir='./logs' # folder do logów
|
logging_dir='./logs' # folder do logów
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue