add bash file

yjh0410 · Aug 22, 2023 · 48ffe12 · 48ffe12
1 parent 50bfbee
commit 48ffe12
Show file tree

Hide file tree

Showing 3 changed files with 60 additions and 0 deletions.
diff --git a/vit_finetune.sh b/vit_finetune.sh
@@ -0,0 +1,21 @@
+# --------------- Finetune on CIFAR ---------------
+python mae_finetune.py \
+        --dataset cifar10 \
+        -m vit_nano \
+        --batch_size 256 \
+        --img_size 32 \
+        --patch_size 2 \
+        --max_epoch 50 \
+        --wp_epoch 5 \
+        --mae_pretrained
+
+# --------------- Finetune on ImageNet ---------------
+# python mae_finetune.py \
+#         --dataset imagenet_1k \
+#         -m vit_nano \
+#         --batch_size 256 \
+#         --img_size 224 \
+#         --patch_size 16 \
+#         --max_epoch 50 \
+#         --wp_epoch 5 \
+#         --mae_pretrained
diff --git a/vit_linprob.sh b/vit_linprob.sh
@@ -0,0 +1,19 @@
+# --------------- Finetune on CIFAR ---------------
+python mae_finetune.py \
+        --dataset cifar10 \
+        -m vit_nano \
+        --batch_size 256 \
+        --img_size 32 \
+        --patch_size 2 \
+        --max_epoch 50 \
+        --wp_epoch 5 \
+
+# --------------- Finetune on ImageNet ---------------
+# python mae_finetune.py \
+#         --dataset imagenet_1k \
+#         -m vit_nano \
+#         --batch_size 256 \
+#         --img_size 224 \
+#         --patch_size 16 \
+#         --max_epoch 50 \
+#         --wp_epoch 5 \
diff --git a/vit_pretrain.sh b/vit_pretrain.sh
@@ -0,0 +1,20 @@
+# --------------- Pretrain on CIFAR ---------------
+python mae_pretrain.py \
+        --dataset cifar10 \
+        -m mae_vit_nano \
+        --batch_size 256 \
+        --img_size 32 \
+        --patch_size 2 \
+        --max_epoch 400 \
+        --wp_epoch 40
+
+
+# --------------- Pretrain on ImageNet ---------------
+# python mae_pretrain.py \
+#         --dataset imagenet_1k \
+#         -m mae_vit_nano \
+#         --batch_size 256 \
+#         --img_size 224 \
+#         --patch_size 16 \
+#         --max_epoch 400 \
+#         --wp_epoch 40