-
Notifications
You must be signed in to change notification settings - Fork 3
/
run.sh
35 lines (28 loc) · 3.93 KB
/
run.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#CIFAR-10 LT (SGD)
python cifar_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule None --dataset cifar10 --log_results --seed 0
python cifar_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule DRW --dataset cifar10 --log_results --seed 0
python cifar_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type LDAM --train_rule DRW --dataset cifar10 --log_results --seed 0
#CIFAR10-LT (SAM)
python cifar_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule None --rho 0.1 --rho_schedule none --log_results --dataset cifar10 --seed 0
python cifar_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule DRW --rho 0.5 --rho_schedule none --log_results --dataset cifar10 --seed 0
python cifar_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type LDAM --train_rule DRW --rho 0.8 --rho_schedule none --log_results --dataset cifar10 --seed 0
#CIFAR100-LT (SGD)
python cifar_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule None --dataset cifar100 --log_results --seed 0
python cifar_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule DRW --dataset cifar10 --log_results --seed 0
python cifar_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type LDAM --train_rule DRW --dataset cifar10 --log_results --seed 0
#CIFAR100-LT (SAM)
python cifar_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule None --rho 0.2 --rho_schedule none --log_results --dataset cifar100 --seed 0
python cifar_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type CE --train_rule DRW --rho 0.8 --rho_schedule none --log_results --dataset cifar100 --seed 0
python cifar_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --loss_type LDAM --train_rule DRW --rho 0.8 --rho_schedule none --log_results --dataset cifar100 --seed 0
#iNaturalist18 (SGD)
python inat_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --data_path inat_18 --loss_type CE --train_rule DRW --dataset inat_18 -b 256 --epochs 90 --arch resnet50 --seed 0 --lr 0.1 --cos_lr --log_results
python inat_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --data_path inat_18 --loss_type LDAM --train_rule DRW --dataset inat_18 -b 256 --epochs 90 --arch resnet50 --seed 0 --lr 0.1 --log_results --cos_lr --margin 0.3 --wd 0.0001
#iNaturalist18 (SAM)
python inat_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --data_path inat_18 --loss_type CE --train_rule DRW --dataset inat_18 -b 256 --epochs 90 --arch resnet50 --cos_lr --rho_schedule step --lr 0.1 --exp_str 0 --seed 1 --rho_steps 0.05 0.1 0.1 0.2 --log_results
python inat_train_sam.py --imb_type exp --imb_factor 0.01 --data_path inat_18 --loss_type LDAM --train_rule DRW --dataset inat_18 -b 256 --epochs 90 --arch resnet50 --cos_lr --gpu 0 --rho_schedule step --lr 0.1 --seed 0 --rho_steps 0.05 0.1 0.5 0.5 --log_results --margin 0.3 --wd 0.0001
#ImageNet-LT (SGD)
python imnet_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --data_path ImageNet --loss_type CE --train_rule DRW --dataset imagenet -b 256 --epochs 90 --arch resnet50 --cos_lr --log_results --lr 0.2 --seed 1
python imnet_train.py --gpu 0 --imb_type exp --imb_factor 0.01 --data_path ImageNet --loss_type LDAM --train_rule DRW --dataset imagenet -b 256 --epochs 90 --arch resnet50 --log_results --seed 0 --wd 2e-4 --lr 0.2 --cos_lr --margin 0.3
#ImageNet-LT (SAM)
python imnet_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --data_path ImageNet --loss_type CE --train_rule DRW --dataset imagenet -b 256 --epochs 90 --arch resnet50 --cos_lr --rho_schedule step --lr 0.2 --seed 0 --rho_steps 0.05 0.1 0.5 0.5 --log_results
python imnet_train_sam.py --gpu 0 --imb_type exp --imb_factor 0.01 --data_path ImageNet --loss_type LDAM --train_rule DRW --dataset imagenet -b 256 --epochs 90 --arch resnet50 --cos_lr --rho_schedule step --lr 0.2 --seed 0 --rho_steps 0.05 0.1 0.5 0.5 --log_results --wd 2e-4 --margin 0.3