From b546f4ee27db9c4bfc6e8cf58940b2f44839f0f2 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Mon, 27 Jan 2025 09:08:39 +0100
Subject: [PATCH] furhter batchsizing downsizing

---
 src/pretrain.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/pretrain.py b/src/pretrain.py
index 38663e3..31dce37 100644
--- a/src/pretrain.py
+++ b/src/pretrain.py
@@ -49,7 +49,7 @@ def pretrain_model(data_path1, data_path2, num_epochs=3):
     aiia_loader = AIIADataLoader(
         merged_df,
         column="image_bytes",
-        batch_size=8,
+        batch_size=4,
         pretraining=True,
         collate_fn=safe_collate
     )