Pixel atari games with DQN

2026-09-09 11:13:47 +08:00 · 2017-05-20 22:09:18 -06:00
parent 7c4919d6ac
commit ad8c8ad853
8 changed files with 151 additions and 50 deletions
@@ -79,7 +79,6 @@ class AsyncAgent:
                episode_steps = 0
                episode_return = 0
                episode += 1
-                policy.update_epsilon()
                terminal = False
                state = task.reset()
                state = state.reshape([1, -1])
@@ -97,6 +96,7 @@ class AsyncAgent:
                state = state.reshape([1, -1])
                value = worker_network.predict(state)
                action = policy.sample(value.flatten())
+                policy.update_epsilon()

            batch_rewards = self.bootstrap_fn(batch_states, batch_actions, batch_rewards,
                                              state, action, terminal, self)