Skip to content

Train a pair of agents to play tennis in Unity's Tennis environment

Notifications You must be signed in to change notification settings

lutaodai/DRL-Tennis-Collaboration

Repository files navigation

DRL-Tennis-Collaboration

Udacity's Deep Reinforcement Learning Nanodegree Project 'Tennis': Training agents to play tennis.

Introduction

For this project, I trained a multi-agent DDPG model to solve the Unity Tennis environment.

In this environment, two agents control rackets to bounce a ball over a net. If an agent hits the ball over the net, it receives a reward of +0.1. If an agent lets a ball hit the ground or hits the ball out of bounds, it receives a reward of -0.01. Thus, the goal of each agent is to keep the ball in play.

The observation space consists of 8 variables corresponding to the position and velocity of the ball and racket. Each agent receives its own, local observation. Two continuous actions are available, corresponding to movement toward (or away from) the net, and jumping.

The task is episodic, and in order to solve the environment, our agents must get an average score of +0.5 (over 100 consecutive episodes, after taking the maximum over both agents). Specifically,

  • After each episode, we add up the rewards that each agent received (without discounting), to get a score for each agent. This yields 2 (potentially different) scores. We then take the maximum of these 2 scores.
  • This yields a single score for each episode.

The environment is considered solved, when the average (over 100 episodes) of those scores is at least +0.5.

Getting Started

To set up your python environment to run the code in this repository, follow the instructions below.

  1. Create (and activate) a new environment with Python 3.6.

    • Linux or Mac:
    conda create --name drlnd python=3.6
    source activate drlnd
    • Windows:
    conda create --name drlnd python=3.6 
    activate drlnd
  2. Clone the repository (if you haven't already!), and navigate to the python/ folder. Then, install several dependencies.

    git clone https://github.com/udacity/deep-reinforcement-learning.git
    cd deep-reinforcement-learning/python
    pip install .
  3. Create an IPython kernel for the drlnd environment.

    python -m ipykernel install --user --name drlnd --display-name "drlnd"
  4. Before running code in a notebook, change the kernel to match the drlnd environment by using the drop-down Kernel menu.

  5. Download the environment from one of the links below. You need only select the environment that matches your operating system:

    (For Windows users) Check out this link if you need help with determining if your computer is running a 32-bit version or 64-bit version of the Windows operating system.

    (For AWS) If you'd like to train the agent on AWS (and have not enabled a virtual screen), then please use this link to obtain the "headless" version of the environment. You will not be able to watch the agent without enabling a virtual screen, but you will be able to train the agent. (To watch the agent, you should follow the instructions to enable a virtual screen, and then download the environment for the Linux operating system above.)

  6. Place the file in the root folder of this GitHub repository, and unzip (or decompress) the file.

  7. If not working in AWS, change file_name in file main.py (row 18) to the correct path to the above env file.

The code provided here was tested on an Ubuntu 16.04 headless (no visual) operating system, but should work for all above provided you follow the correct instructions.

Setup of repository

Apart from the Readme.md file this repository consists of the following files:

  1. config.py: Configuration files for training the model;
  2. model.py: Actor and Critc Network classes;
  3. ddpg_agent.py: Agent, ReplayBuffer and OUNoise classes; The Agent class makes use of the Actor and Critic classes from model.py, the ReplayBuffer class and the OUNoise class;
  4. multi_agents.py: MultiAgent class defining multiple agents based on the Agent class;
  5. run.py: Script which will train the agent. Can be run directly from the terminal;
  6. checkpoint_[01].actor.pth: Contains the weights of successful Actor Networks;
  7. checkpoint_[01].critic.pth: Contains the weights of successful Critic Networks.

Todos:

  1. report.ipynb: As an alternative to the run.py script this Jupyter Notebook has a step-by-step structure. Here the learning algorithm is described in detail;
  2. Improve plots;
  3. Add more screen outputs during the training to better monitor the learning;
  4. Add specific dirs for saving plots and model weights;
  5. Add functions of loading weights and resume training;
  6. Add gif visualizing the performance of the trained model;

Instruction

To train the model, simply adjust parameters in the config.py file, and then run

python run.py

About

Train a pair of agents to play tennis in Unity's Tennis environment

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages