README This program trains a Dueling DQN on Atari Games. Currently it supports Breakout. Later we can easily add support for other games too. To run it, first install these dependencies:
- Tensorflow 2.3.0 (GPU)
- OpenCV
- OpenAI Gym
- Gym support for Atari
After installing these you can just
- For training the model run python3 train_dqn.py. In case you are trying to train from a previous checkpoint add a cmdline argument for its location, e.g python3 train_dqn.py saves/save-000/.
- For evaluating a trained model just run python3 evaluation.py location_of_model.
The files in the folder contain code for:
- agent_dueling_dqn.py: Class for Dueling DQN
- AtariWrapper: Wrapper Class for Gym[Atari].
- Config.py: File for global constants. You can modify those variables. Some variables are hyperparameters for further tuning.
- Evaluation.py: Code for testing your model in a game visually!.
- ReplayBuffer.py: Class for replayBuffer. In future will support prioritized experience replay.
- train_dqn.py: Main file for training the model.