-
Notifications
You must be signed in to change notification settings - Fork 1
/
start_training.sh
executable file
·67 lines (63 loc) · 3.45 KB
/
start_training.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
#!/bin/bash
ipadress=$(hostname -I | cut -d' ' -f1)
number=${ipadress: -3}
case $number in
120)
echo -n "120"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
119)
echo -n "119"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
118)
echo -n "118"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 2000 --gamma 0.99 --level_files tunnel.svg"
;;
117)
echo -n "117"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 2000 --gamma 0.99 --level_files tunnel.svg"
;;
116)
echo -n "116"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
115)
echo -n "115"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
114)
echo -n "114"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 8 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
113)
echo -n "113"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 256 --sync_experience 256 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
112)
echo -n "112"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 8 --log_interval 10 --steps 2000 --gamma 0.99 --level_files tunnel.svg"
;;
173)
echo -n "173"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 256 --sync_experience 256 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 2000 --gamma 0.99 --level_files tunnel.svg"
;;
110)
echo -n "110"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 128 --sync_experience 128 --batch_size 1 --K_epochs 8 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
109)
echo -n "109"
args="--model_name model --ckpt_folder ./models/$number --mode train --update_experience 256 --sync_experience 256 --batch_size 1 --K_epochs 4 --log_interval 10 --steps 10000 --gamma 0.99 --level_files svg3_tareq.svg"
;;
*)
echo -n "du dumm"
;;
esac
cd ~/Dokumente/SauRoN/
mkdir ./models/$number
echo "$args" > ./models/$number/args.txt
source "/home/nex/anaconda3/etc/profile.d/conda.sh"
export PATH="/home/nex/anaconda3/bin:$PATH"
conda activate sauron
python main.py $args