-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathtrain-office-home-ood_is_test.sh
executable file
·10 lines (8 loc) · 1.54 KB
/
train-office-home-ood_is_test.sh
1
2
3
4
5
6
7
8
9
10
# ERM
python ./src/main.py --ensemble_size 5 --batch_size_train 128 --batch_size_eval 512 --seed 0 --epochs 600 --lr 0.001 --opt sgd --scheduler none --alpha 0 --perturb_type ood_is_test --model resnet18 --dataset oh-65cls --eval_freq 1000 --l2_reg 0.0001 --no_diversity
python ./src/main.py --ensemble_size 5 --batch_size_train 128 --batch_size_eval 512 --seed 1 --epochs 600 --lr 0.001 --opt sgd --scheduler none --alpha 0 --perturb_type ood_is_test --model resnet18 --dataset oh-65cls --eval_freq 1000 --l2_reg 0.0001 --no_diversity
python ./src/main.py --ensemble_size 5 --batch_size_train 128 --batch_size_eval 512 --seed 2 --epochs 600 --lr 0.001 --opt sgd --scheduler none --alpha 0 --perturb_type ood_is_test --model resnet18 --dataset oh-65cls --eval_freq 1000 --l2_reg 0.0001 --no_diversity
# D-BAT
python ./src/main.py --ensemble_size 5 --batch_size_train 128 --batch_size_eval 512 --seed 0 --epochs 600 --lr 0.001 --opt sgd --scheduler none --alpha 0.00001 --perturb_type ood_is_test --model resnet18 --dataset oh-65cls --eval_freq 1000 --l2_reg 0.0001
python ./src/main.py --ensemble_size 5 --batch_size_train 128 --batch_size_eval 512 --seed 1 --epochs 600 --lr 0.001 --opt sgd --scheduler none --alpha 0.00001 --perturb_type ood_is_test --model resnet18 --dataset oh-65cls --eval_freq 1000 --l2_reg 0.0001
python ./src/main.py --ensemble_size 5 --batch_size_train 128 --batch_size_eval 512 --seed 2 --epochs 600 --lr 0.001 --opt sgd --scheduler none --alpha 0.00001 --perturb_type ood_is_test --model resnet18 --dataset oh-65cls --eval_freq 1000 --l2_reg 0.0001