Deep Q Networks
Actions space: Discrete
References: Playing Atari with Deep Reinforcement Learning
Training the network
- Sample a batch of transitions from the replay buffer.
- Using the next states from the sampled batch, run the target network to calculate the values for each of the actions , and keep only the maximum value for each state.
- In order to zero out the updates for the actions that were not played (resulting from zeroing the MSE loss), use the current states from the sampled batch, and run the online network to get the current Q values predictions. Set those values as the targets for the actions that were not actually played.
For each action that was played, use the following equation for calculating the targets of the network:
Finally, train the online network using the current states as inputs, and with the aforementioned targets.
- Once in every few thousand steps, copy the weights from the online network to the target network.