bs

mwalmsley · Nov 5, 2023 · b011cd9 · b011cd9
1 parent 4941c15
commit b011cd9
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,6 +21,6 @@ export NCCL_BLOCKING_WAIT=1  #Set this environment variable if you wish to use t
 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