4gpu b0 why not

mwalmsley · Nov 23, 2023 · e381992 · e381992
1 parent 8668f4c
commit e381992
Showing 1 changed file with 2 additions and 2 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? - 20%, could do x4 -> 256
 
 srun $PYTHON $REPO_DIR/only_for_me/narval/train.py \
-    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_pit_xs_224_b256_4gpu \
+    --save-dir $REPO_DIR/only_for_me/narval/desi_300px_efficientnet_b0_4gpu \
     --batch-size 256 \
     --gpus 4 \
     --num-workers 10 \
-    --architecture pit_xs_224 \
+    --architecture efficientnet_b0 \
     --color --wandb --mixed-precision --compile-encoder
 
     # maxvit_small_tf_224 \