reduce batch size

2026-06-22 13:38:06 +02:00
parent 164e25fe55
commit b2d553c20d
1 changed files with 2 additions and 2 deletions
@@ -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