smaller batch, still 11 workers

mwalmsley · Nov 6, 2023 · 666241d · 666241d
1 parent d1e6fa5
commit 666241d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -21,7 +21,7 @@ mkdir /tmp/cache
 REPO_DIR=/project/def-bovy/walml/zoobot/
 srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
     --save-dir $REPO_DIR/only_for_me/narval/debug_models \
-    --batch-size 512 \
+    --batch-size 256 \
     --color --wandb --mixed-precision
 # srun python $SLURM_TMPDIR/zoobot/only_for_me/narval/finetune.py