-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathrunner.sh
25 lines (20 loc) · 840 Bytes
/
runner.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
# basic info
id=CIFAR-100
ood="SVHN places365 LSUN iSUN dtd"
# training info
batch=512
epochs=500
lr=0.5
wd=1e-6
backbone=resnet34
pcon=1.
m=0.999
# Runing PALM on supervised OOD detection
k=5
cache=6
method=top$k-palm-cache$cache-ema$m
save_path=checkpoints/$id-$backbone-$method.pt
python main.py --in-dataset $id --backbone $backbone --method $method --epochs $epochs --save-path $save_path -b $batch --lr $lr --wd $wd --cache-size $cache --lambda_pcon $pcon --proto_m $m --k $k
python feature_extract.py --in-dataset $id --out-datasets $ood --backbone $backbone --method $method --epochs $epochs --save-path $save_path --cache-size $cache
score="mahalanobis"
python eval_cifar.py --in-dataset $id --out-datasets $ood --backbone $backbone --method $method --epochs $epochs --save-path $save_path --score $score --cache-size $cache