Skip to content

Train a MADDPG algorithm to solve a tennis environment.

Notifications You must be signed in to change notification settings

pestipeti/UdacityDRLTennis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep RL - Collaboration and Competition

  • Organization: Udacity
  • Course: Deep Reinforcement Learning NanoDegree.
  • Project: #3 - Collaboration and Competition

Project Details

For this project, we will work with the Tennis environment.

(Click on the image to watch the video)

Trained Agents

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.

Solving the Environment

The task is episodic, and in order to solve the environment, your 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

  1. Clone this repository

  2. 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 (version 1) or this link (version 2) 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.)

  3. Place the file in the root folder of the cloned repository, and unzip (or decompress) the file.

Instructions

Code

UdacityDRLContinuousControl/
  |- checkpoints/
  |        |- checkpoint_0_actor.pth
  |        |- checkpoint_0_critic.pth
  |        |- checkpoint_1_actor.pth
  |        |- checkpoint_1_critic.pth
  |- .gitignore
  |- agent.py
  |- maddpg.py
  |- model.py
  |- README.md
  |- REPORT.md
  |- test.py
  |- train.py
  |- train_scores.png
  • checkpoints/ - Actor/critic wegiths for both agents
  • .gitignore - Gitignore file
  • agent.py - Navigation agent, interacts with and learns from the environment.
  • maddpg.py - Multi Agent Deep Deterministic Policy Gradient (train and test code)
  • model.py - Actor and critic models
  • README.md - this readme file
  • REPORT.md - Implementation description
  • test.py - Test the agent
  • train.py - Train the agent
  • train_scores.png - Training progress plot

Train

$ python train.py

Test

$ python test.py

Results

Training progress of DQN model

About

Train a MADDPG algorithm to solve a tennis environment.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages