This commit is contained in:
l.gabrysiak 2025-02-25 14:11:46 +01:00
parent 5c0e4381e7
commit 0c01db5995
1 changed files with 2 additions and 2 deletions

4
hft.py
View File

@ -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
)