From b3d1e0295af07d72255c2243d23f803b26d7603e Mon Sep 17 00:00:00 2001
From: "l.gabrysiak" <l.gabrysiak@osadkowski.pl>
Date: Tue, 25 Feb 2025 13:54:21 +0100
Subject: [PATCH] mod

---
 hft.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/hft.py b/hft.py
index eccec6c..fbcbd8c 100644
--- a/hft.py
+++ b/hft.py
@@ -178,7 +178,11 @@ training_args = TrainingArguments(
     save_strategy="steps",
     save_steps=1000,
     report_to="none",
-    gradient_checkpointing=True
+    gradient_checkpointing=True,
+    evaluation_strategy="epoch",     # częstotliwość oceny
+    per_device_train_batch_size=8,   # batch size dla treningu
+    per_device_eval_batch_size=8,    # batch size dla ewaluacji
+    logging_dir='./logs'             # folder do logów
 )
 
 # Trening
@@ -186,8 +190,7 @@ trainer = CustomTrainer(
     model=model,
     args=training_args,
     train_dataset=tokenized_dataset,
-    data_collator=custom_collate_fn,  # Użyj niestandardowego collate_fn
-    batch_size=8  # zmniejszenie rozmiaru batcha
+    data_collator=custom_collate_fn  # Użyj niestandardowego collate_fn
 )
 trainer.train()