maxvit_rmlp_base_rw_224

mwalmsley · Nov 24, 2023 · 5854015 · 5854015
1 parent b979139
commit 5854015
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/only_for_me/narval/train.sh b/only_for_me/narval/train.sh
@@ -45,13 +45,14 @@ REPO_DIR=/project/def-bovy/walml/zoobot
 # convnext_small 64 - 49.25%, MAYBE 128 
 # efficientnet_b4 - 48% at 64, could maybe do 128
 # efficientnet_b5 - 64?
+# maxvit_rmlp_base_rw_224 - 32?
 
 srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
-    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_efficientnet_b5_4gpu \
-    --batch-size 64 \
+    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_maxvit_rmlp_base_rw_224_4gpu \
+    --batch-size 32 \
     --gpus 4 \
     --num-workers 10 \
-    --architecture efficientnet_b5 \
+    --architecture maxvit_rmlp_base_rw_224 \
     --color --wandb --mixed-precision --compile-encoder
 
     # maxvit_small_tf_224 \