Commit 94c429ce authored by lli's avatar lli
Browse files

update dqn params

parent 89c9da8d
......@@ -16,11 +16,7 @@ n_action = env.action_space.n # Number of output
#lr = 0.001 # Learning rate
gamma = 1 # Discount factor
epsilon = 1 # epsilon start value
epsilon_end = 0.001 # epsilon end value
epsilon_decay = True # epsilon decay rate
decay = 50
steps_done = 0 # parameter for epsilon greedy policy
replay_buffer = deque(maxlen=10000) # Size of replay buffer
replay_batch_size = 64 # Size of replay batch
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment