-
Notifications
You must be signed in to change notification settings - Fork 11
/
BFA_CIFAR.sh
78 lines (67 loc) · 2.1 KB
/
BFA_CIFAR.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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
#!/usr/bin/env sh
############### Host ##############################
HOST=$(hostname)
echo "Current host is: $HOST"
# Automatic check the host and configure
case $HOST in
"alpha")
PYTHON="/home/elliot/anaconda3/envs/pytorch041/bin/python" # python environment path
TENSORBOARD='/home/elliot/anaconda3/envs/pytorch041/bin/tensorboard' # tensorboard environment path
data_path='/home/elliot/data/pytorch/cifar10'
;;
esac
DATE=`date +%Y-%m-%d`
if [ ! -d "$DIRECTORY" ]; then
mkdir ./save/${DATE}/
fi
############### Configurations ########################
enable_tb_display=false # enable tensorboard display
model=resnet20_quan
dataset=cifar10
test_batch_size=128
label_info=BFA_defense_test
attack_sample_size=128 # number of data used for BFA
n_iter=100 # number of iteration to perform BFA
k_top=100 # only check k_top weights with top gradient ranking in each layer
save_path=./save/${DATE}/${dataset}_${model}_${label_info}
tb_path=${save_path}/tb_log #tensorboard log path
# set the pretrained model path
pretrained_model=/home/elliot/Documents/CVPR_2020/BFA_defense_camera_ready/BFA/save/2020-05-08/cifar10_resnet20_quan_160_SGD_binarized/model_best.pth.tar
############### Neural network ############################
COUNTER=0
{
while [ $COUNTER -lt 1 ]; do
$PYTHON main.py --dataset ${dataset} \
--data_path ${data_path} \
--arch ${model} --save_path ${save_path} \
--test_batch_size ${test_batch_size} --workers 8 --ngpu 1 --gpu_id 1 \
--print_freq 50 \
--evaluate --resume ${pretrained_model} --fine_tune\
--reset_weight --bfa --n_iter ${n_iter} \
--attack_sample_size ${attack_sample_size} \
let COUNTER=COUNTER+1
done
} &
############## Tensorboard logging ##########################
{
if [ "$enable_tb_display" = true ]; then
sleep 30
wait
$TENSORBOARD --logdir $tb_path --port=6006
fi
} &
{
if [ "$enable_tb_display" = true ]; then
sleep 45
wait
case $HOST in
"Hydrogen")
firefox http://0.0.0.0:6006/
;;
"alpha")
google-chrome http://0.0.0.0:6006/
;;
esac
fi
} &
wait