-
Notifications
You must be signed in to change notification settings - Fork 3
/
train.sh
19 lines (13 loc) · 954 Bytes
/
train.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#!/bin/bash
SEED=0
python main.py --mode train -g default -e cmpae_s${SEED} -w --seed ${SEED} --gpu 0
echo "Begin to select thresholds from the validation set."
echo "Note: In this stage, five processes will run in parallel to save time."
echo "Please wait for about 10 min."
python main.py --mode select_thresholds -g default -e cmpae_s${SEED} --seed ${SEED} -w -s --gpu 0 --start_class 0 &
python main.py --mode select_thresholds -g default -e cmpae_s${SEED} --seed ${SEED} -w -s --gpu 1 --start_class 5 &
python main.py --mode select_thresholds -g default -e cmpae_s${SEED} --seed ${SEED} -w -s --gpu 2 --start_class 10 &
python main.py --mode select_thresholds -g default -e cmpae_s${SEED} --seed ${SEED} -w -s --gpu 3 --start_class 15 &
python main.py --mode select_thresholds -g default -e cmpae_s${SEED} --seed ${SEED} -w -s --gpu 4 --start_class 20 &
wait
python main.py --mode test -g default -e cmpae_s${SEED} --seed ${SEED} -w -s --gpu 0