From 23ca029702a02dcb4c6d6d57155a6c140f63890c Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Sun, 5 Nov 2023 19:20:10 -0500 Subject: [PATCH] b512 --- only_for_me/narval/train.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh index 320eb96d..a10dad29 100644 --- 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 256 \ + --batch-size 512 \ --color --wandb --mixed-precision # srun python $SLURM_TMPDIR/zoobot/only_for_me/narval/finetune.py