An implementation of Synchronous Advantage Actor Critic (A2C) in TensorFlow. A2C is a variant of advantage actor critic introduced by OpenAI in their published baselines. However, these baselines are difficult to understand and modify. So, I made the A2C based on their implementation but in a clearer and simpler way.
- Support for Tensorboard visualization per running agent in an environment.
- Support for different policy networks in an easier way.
- Support for environments other than OpenAI gym in an easy way.
- Support for video generation of an agent acting in the environment.
- Simple and easy code to modify and begin experimenting. All you need to do is plug and play!
Asynchronous advantage actor critic was introduced in Asynchronous Methods for Deep Reinforcement Learning. The difference between both methods is that in asynchronous AC, parallel agents update the global network each one on its own. So, at a certain time, the weights used by an agent maybe different than the weights used by another agent leading to the fact that each agent plays with a different policy to explore more and more of the environment. However, in synchronous AC, all of the updates by the parallel agents are collected to update the global network. To encourage exploration, stochastic noise is added to the probability distribution of the actions predicted by each agent.
This implementation allows for using different environments. It's not restricted to OpenAI gym environments. If you want to attach the project to another environment rather than that provided by gym, all you have to do is to inherit from the base class BaseEnv in envs/base_env.py, and implement all the methods in a plug and play fashion (See the gym environment example class). You also have to add the name of the new environment class in A2C.py\env_name_parser() method.
The methods that should be implemented in a new environment class are:
make()for creating the environment and returning a reference to it.step()for taking a step in the environment and returning a tuple (observation images, reward float value, done boolean, any other info).reset()for resetting the environment to the initial state.get_observation_space()for returning an object with attribute tupleshaperepresenting the shape of the observation space.get_action_space()for returning an object with attributenrepresenting the number of possible actions in the environment.render()for rendering the environment if appropriate.
This implementation comes with the basic CNN policy network from OpenAI baseline. However, it supports using different policy networks. All you have to do is to inherit from the base class BasePolicy in models\base_policy.py, and implement all the methods in a plug and play fashion again :D (See the CNNPolicy example class). You also have to add the name of the new policy network class in models\model.py\policy_name_parser() method.
This implementation allows for the beautiful Tensorboard visualization. It displays the time plots per running agent of the two most important signals in reinforcement learning: episode length and total reward in the episode. All you have to do is to launch Tensorboard from your experiment directory located in experiments/.
tensorboard --logdir=experiments/my_experiment/summaries
During training, you can generate videos of the trained agent acting (playing) in the environment. This is achieved by changing record_video_every in the configuration file from -1 to the number of episodes between two generated videos. Videos are generated in your experiment directory.
During testing, videos are generated automatically if the optional monitor method is implemented in the environment. As for the gym included environment, it's already been implemented.
Python 3 or above
tensorflow 1.3.0
numpy 1.13.1
gym 0.9.2
tqdm 4.15.0
bunch 1.0.1
matplotlib 2.0.2
Pillow 4.2.1
python main.py config/test.json
The file 'test.json' is just an example of a file having all parameters to train on environments. You can create your own configuration file for training/testing.
In the project, two configuration files are provided as examples for training on Pong and Breakout Atari games.
| Model | Game | Average Score | Max Score |
|---|---|---|---|
| CNNPolicy | Pong | 17 | 21 |
| CNNPolicy | Breakout | 650 | 850 |
- Inference and training are working properly.
This project is licensed under the Apache License 2.0 - see the LICENSE file for details.