update configs

Yuyan-C · Mar 15, 2023 · 1e0fb9d · 1e0fb9d
1 parent 4051103
commit 1e0fb9d
Show file tree

Hide file tree

Showing 8 changed files with 37 additions and 14 deletions.
diff --git a/scripts/run_aircraft.sh b/scripts/run_aircraft.sh
@@ -3,7 +3,7 @@
 set -e
 set -x
 
-CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --dataset_name 'aircraft' \
     --batch_size 128 \
     --grad_from_block 11 \
@@ -18,5 +18,5 @@ CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
     --warmup_teacher_temp 0.07 \
     --teacher_temp 0.04 \
     --warmup_teacher_temp_epochs 30 \
-    --memax_weight 2 \
+    --memax_weight 1 \
     --exp_name aircraft_simgcd
diff --git a/scripts/run_cars.sh b/scripts/run_cars.sh
@@ -3,7 +3,7 @@
 set -e
 set -x
 
-CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --dataset_name 'scars' \
     --batch_size 128 \
     --grad_from_block 11 \
@@ -18,5 +18,5 @@ CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
     --warmup_teacher_temp 0.07 \
     --teacher_temp 0.04 \
     --warmup_teacher_temp_epochs 30 \
-    --memax_weight 2 \
+    --memax_weight 1 \
     --exp_name scars_simgcd
diff --git a/scripts/run_cifar10.sh b/scripts/run_cifar10.sh
@@ -3,7 +3,7 @@
 set -e
 set -x
 
-CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --dataset_name 'cifar10' \
     --batch_size 128 \
     --grad_from_block 11 \
@@ -18,5 +18,5 @@ CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
     --warmup_teacher_temp 0.07 \
     --teacher_temp 0.04 \
     --warmup_teacher_temp_epochs 30 \
-    --memax_weight 2 \
+    --memax_weight 1 \
     --exp_name cifar10_simgcd
diff --git a/scripts/run_cifar100.sh b/scripts/run_cifar100.sh
@@ -3,7 +3,7 @@
 set -e
 set -x
 
-CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --dataset_name 'cifar100' \
     --batch_size 128 \
     --grad_from_block 11 \
@@ -18,5 +18,5 @@ CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
     --warmup_teacher_temp 0.07 \
     --teacher_temp 0.04 \
     --warmup_teacher_temp_epochs 30 \
-    --memax_weight 2 \
+    --memax_weight 4 \
     --exp_name cifar100_simgcd
diff --git a/scripts/run_cub.sh b/scripts/run_cub.sh
@@ -3,7 +3,7 @@
 set -e
 set -x
 
-CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --dataset_name 'cub' \
     --batch_size 128 \
     --grad_from_block 11 \

diff --git a/scripts/run_herb19.sh b/scripts/run_herb19.sh
@@ -3,7 +3,7 @@
 set -e
 set -x
 
-CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --dataset_name 'herbarium_19' \
     --batch_size 128 \
     --grad_from_block 11 \
@@ -14,9 +14,9 @@ CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
     --weight_decay 5e-5 \
     --transform 'imagenet' \
     --lr 0.1 \
-    --eval_funcs 'v2' 'v2p' \
+    --eval_funcs 'v2' 'v2b' \
     --warmup_teacher_temp 0.07 \
     --teacher_temp 0.04 \
     --warmup_teacher_temp_epochs 30 \
-    --memax_weight 2 \
+    --memax_weight 1 \
     --exp_name herb19_simgcd
diff --git a/scripts/run_imagenet100.sh b/scripts/run_imagenet100.sh
@@ -3,7 +3,7 @@
 set -e
 set -x
 
-CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
+CUDA_VISIBLE_DEVICES=0 python train.py \
     --dataset_name 'imagenet_100' \
     --batch_size 128 \
     --grad_from_block 11 \
@@ -18,5 +18,5 @@ CUDA_VISIBLE_DEVICES=0 python -m model.simgcd \
     --warmup_teacher_temp 0.07 \
     --teacher_temp 0.04 \
     --warmup_teacher_temp_epochs 30 \
-    --memax_weight 2 \
+    --memax_weight 1 \
     --exp_name imagenet100_simgcd
diff --git a/scripts/run_imagenet1k.sh b/scripts/run_imagenet1k.sh
@@ -0,0 +1,23 @@
+#!/bin/bash
+
+set -e
+set -x
+
+CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --master_port 12348 --nproc_per_node=8 train_mp.py \
+    --dataset_name 'imagenet_1k' \
+    --batch_size 128 \
+    --grad_from_block 11 \
+    --epochs 200 \
+    --num_workers 8 \
+    --use_ssb_splits \
+    --sup_weight 0.35 \
+    --weight_decay 5e-5 \
+    --transform 'imagenet' \
+    --lr 0.1 \
+    --eval_funcs 'v2' \
+    --warmup_teacher_temp 0.07 \
+    --teacher_temp 0.04 \
+    --warmup_teacher_temp_epochs 30 \
+    --memax_weight 1 \
+    --exp_name imagenet1k_simgcd \
+    --print_freq 100