From 0c01db5995cd4a70f77feb17796e316348749019 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Tue, 25 Feb 2025 14:11:46 +0100 Subject: [PATCH] mod --- hft.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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 )