furhter batchsizing downsizing

2025-01-27 09:08:39 +01:00 · 2025-01-27 09:08:39 +01:00 · b546f4ee27
parent 8dad1d7150
commit b546f4ee27
1 changed files with 1 additions and 1 deletions
--- 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
    )