pit s 64

mwalmsley · Nov 23, 2023 · e1aebaf · e1aebaf
1 parent 94d1d5d
commit e1aebaf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -36,11 +36,11 @@ REPO_DIR=/project/def-bovy/walml/zoobot
 # pit_s_224 64?
 
 srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
-    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_pit_xs_224_1gpu \
-    --batch-size 256 \
+    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_pit_s_224_1gpu \
+    --batch-size 64 \
     --gpus 1 \
     --num-workers 10 \
-    --architecture pit_xs_224 \
+    --architecture pit_s_224 \
     --color --wandb --mixed-precision --compile-encoder
 
     # maxvit_small_tf_224 \