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,
|
||||
report_to="none",
|
||||
gradient_checkpointing=True,
|
||||
per_device_train_batch_size=8, # batch size dla treningu
|
||||
per_device_eval_batch_size=8, # batch size dla ewaluacji
|
||||
per_device_train_batch_size=4, # batch size dla treningu
|
||||
per_device_eval_batch_size=4, # batch size dla ewaluacji
|
||||
logging_dir='./logs' # folder do logów
|
||||
)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue