-
Notifications
You must be signed in to change notification settings - Fork 2
/
run_lider.sh
executable file
·36 lines (33 loc) · 1.08 KB
/
run_lider.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
#!/bin/bash
# Six games to choose from
game=$1
if [ "$game" == "Gopher" ]; then
echo "Training Gopher..."
elif [ "$game" == "NameThisGame" ]; then
echo "Training NameThisGame..."
elif [ "$game" == "MsPacman" ]; then
echo "Training Ms.Pac-Man..."
elif [ "$game" == "Alien" ]; then
echo "Training Alien..."
elif [ "$game" == "Freeway" ]; then
echo "Training Freeway..."
elif [ "$game" == "MontezumaRevenge" ]; then
echo "Training Montezuma's Revenge..."
else
echo "invalid game!"
echo "Choose from [Gopher, NameThisGame, MsPacman, Alien, Freeway, MontezumaRevenge]."
echo "For example: ./run_lider.sh MsPacman"
echo "Exiting..."; exit
fi
python3 LiDER/run_experiment.py \
--gym-env=${game}NoFrameskip-v4 \
--parallel-size=17 \
--max-time-step-fraction=0.5 \
--use-mnih-2015 --input-shape=88 --padding=SAME \
--unclipped-reward --transformed-bellman \
--use-sil --priority-memory \
--batch-size=32 \
--checkpoint-buffer --checkpoint-freq=1 \
--append-experiment-num=1 \
--use-lider \
# --use-gpu --cuda-devices=0 \ # uncomment for gpu support