diff --git a/hft.py b/hft.py index f83d558..cc4f09d 100644 --- a/hft.py +++ b/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 )