forked from Naresh1318/Playing_Atari_the_right_way
-
Notifications
You must be signed in to change notification settings - Fork 0
/
mission_control_breakout_ram.py
34 lines (28 loc) · 1021 Bytes
/
mission_control_breakout_ram.py
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
########################################################################################################################
# Training
learning_rate = 0.00015
batch_size = 32
observation_time = int(1e6) # 1e3
rand_observation_time = int(5e4) # 5e2
target_network_update = int(1e4) # 1e3
prob_random = 1
gamma = 0.99
n_episodes = int(2e4) # 10
fit_epochs = 1
weight_init = 0.01
momentum = 0.95
epsilon = 0.01
########################################################################################################################
# Agent Model
dense_1 = 512
dense_2 = 512
dense_3 = 256
########################################################################################################################
# Control
train_model = False
load_trained_model = True
show_ui = True
show_action = False
########################################################################################################################
# Paths
logdir = "./Results/Breakout_ram/" # Use: "./Results/CartPole/", "./Results/Breakout/"