typo

mwalmsley · Nov 30, 2023 · 49db7f5 · 49db7f5
1 parent 1e5fcc7
commit 49db7f5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -26,8 +26,9 @@ REPO_DIR=/project/def-bovy/walml/zoobot
 #     --color --wandb --mixed-precision 
 
 srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
-    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_maxvittiny_1gpu \
+    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_maxvit_tiny_rw_224_1gpu \
     --batch-size 64 \
+    --architecture maxvit_tiny_rw_224 \
     --gpus 1 \
     --num-workers 10 \
     --color --wandb --mixed-precision 
@@ -38,7 +39,7 @@ srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
 # batch sizes
 # v100
 # efficientnet_b0 256
-# maxvittiny_rw_224 64
+# maxvit_tiny_rw_224 64
 # tf_efficientnetv2_b0 256 - 50.55%, might squeeze x2
 # tf_efficientnetv2_s 64? TODO
 # pit_xs_224 512