From b2d553c20d8d12e4fa9338f277ee4f4e4854d39b Mon Sep 17 00:00:00 2001
From: Yann Ahlgrim <yann@yann-x1carbon.ahlgrim.net>
Date: Mon, 22 Jun 2026 13:38:06 +0200
Subject: [PATCH] reduce batch size

---
 configs/grids/lp_grid.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/grids/lp_grid.yaml b/configs/grids/lp_grid.yaml
index 0960cf1..99fb0a9 100644
--- a/configs/grids/lp_grid.yaml
+++ b/configs/grids/lp_grid.yaml
@@ -10,13 +10,13 @@ grid:
   optimization.momentum: [0.9]
   optimization.lr_schedule: [cosine]
   optimization.gradient_accumulation_steps: [8, 16]
-  data.batch_size: [64]
+  data.batch_size: [16]
   meta.representation_type: [last_avgpool, last4_avgpool_concat]
   meta.head_type: [linear, bn_linear]
 
 launch:
   folder: submitit_logs/
-  partition: gpu8
+  partition: gpu1
   nodes: 1
   tasks_per_node: 1
   time: 180