-
Notifications
You must be signed in to change notification settings - Fork 2
/
multiruntinyimagenet_jda.sh
executable file
·21 lines (16 loc) · 1.17 KB
/
multiruntinyimagenet_jda.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/bin/sh
CUDA_VISIBLE_DEVICES=0 python C100/train_student_fakd.py --checkpoint-dir ./store/ \
--gpu-id 0 --i 1 --data /home/sst/dataset/c100 --arch wrn_16_2_fakd --tarch wrn_40_2 \
--tcheckpoint checkpoints/wrn_40_2.pth --dirac_ratio 1.0
CUDA_VISIBLE_DEVICES=0 python C100/train_student_fakd.py --checkpoint-dir ./store/ \
--gpu-id 0 --i 1 --data /home/sst/dataset/c100 --arch wrn_16_2_fakd --tarch wrn_40_2 \
--tcheckpoint checkpoints/wrn_40_2.pth --dirac_ratio 0.75
CUDA_VISIBLE_DEVICES=0 python C100/train_student_fakd.py --checkpoint-dir ./store/ \
--gpu-id 0 --i 1 --data /home/sst/dataset/c100 --arch wrn_16_2_fakd --tarch wrn_40_2 \
--tcheckpoint checkpoints/wrn_40_2.pth --dirac_ratio 0.0
CUDA_VISIBLE_DEVICES=0 python C100/train_student_fakd.py --checkpoint-dir ./store/ \
--gpu-id 0 --i 1 --data /home/sst/dataset/c100 --arch wrn_16_2_fakd --tarch wrn_40_2 \
--tcheckpoint checkpoints/wrn_40_2.pth --dirac_ratio 0.25
CUDA_VISIBLE_DEVICES=0 python C100/train_student_fakd.py --checkpoint-dir ./store/ \
--gpu-id 0 --i 1 --data /home/sst/dataset/c100 --arch wrn_16_2_fakd --tarch wrn_40_2 \
--tcheckpoint checkpoints/wrn_40_2.pth --dirac_ratio 0.5