-
Notifications
You must be signed in to change notification settings - Fork 0
/
RUNME.sh
55 lines (45 loc) · 1.38 KB
/
RUNME.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
#! /bin/bash
export DISPLAY=':0.0'
#set experimental configuration
CI='python -d -Wonce -Werror'
gammas=(0 1 2)
confs=('conf/test.yaml' 'conf/mnist.yaml')
device='.cache/nvidia-smi-x.xml'
time=$(date +%s)
#clear output and cache
cat /dev/null > ./nohup.out
rm -rf ./.cache
#build folder for this experiment
mkdir -p .cache/pretrained/$time
mkdir -p .cache/trained/$time
#training
for conf in ${confs[@]}
do
echo ================ pretrain for $conf ================ ;
nvidia-smi -q -x -f $device ;
$CI pretrain.py --conf $conf --device $device --time $time || break ;
for gamma in ${gammas[@]} ;
do
echo ================ train gamma = $gamma for $conf ================ ;
nvidia-smi -q -x -f $device ;
$CI train.py --gamma $gamma --conf $conf --device $device --time $time ;
done
echo -------------------------------------- ;
echo ---------------- done ---------------- ;
echo -------------------------------------- ;
done
#evaluating
nvidia-smi -q -x -f $device
for conf in ${confs[@]}
do
for gamma in ${gammas[@]} ;
do
echo ================ evaluate gamma = $gamma for $conf ================ ;
$CI eval.py --gamma $gamma --conf $conf --device $device --time $time ;
done
echo -------------------------------------- ;
echo ---------------- done ---------------- ;
echo -------------------------------------- ;
done
#if pause in the end is required, uncomment the next line
#read