forked from stone8oy/deepRL
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_gpu.sh
37 lines (31 loc) · 1.25 KB
/
run_gpu.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
#!/bin/bash
#program options
GPU=true
GUI=false
rom=./roms/breakout.bin
epsilon_start=1
epsilon_min=0.1
epsilon_decay=0.99
epsilon_explore_idx=1000
replay_memory_capacity=100000
replay_start_size=10000
sampleStrategy="uniform"
update_frequency=4
discount_factor=0.98
solver=./prototxt/aleSolver.prototxt
skip_frame=4
model=./models/rmsprop_nature_iter_100000.caffemodel
evaluate=false
doubleQ=true
eval_epsilon=0.05
target_q_freq=1000
args="--gpu ${GPU} --gui ${GUI} --rom ${rom} --doubleQ ${doubleQ} --target_q_freq ${target_q_freq} --eval_epsilon ${eval_epsilon} --epsilon_start ${epsilon_start} --epsilon_min ${epsilon_min} --epsilon_decay ${epsilon_decay} --epsilon_explore_idx ${epsilon_explore_idx} --replay_memory_capacity ${replay_memory_capacity} --replay_start_size ${replay_start_size} --sampleStrategy ${sampleStrategy} --update_frequency ${update_frequency} --discount_factor ${discount_factor} --solver ${solver} --skip_frame ${skip_frame} --model ${model} --evaluate ${evaluate}"
echo ${args}
# compile project deepRL
echo "=============================>Compling project: deepRL"
rm CMakeCache.txt
cp CMakeLists.gpu.txt CMakeLists.txt
cmake .
make clean & make -j2 2> build.errlog
#run , you need set solver_mode: GPU in your solver.prototxt
./deepRL ${args}