Showcase of DDPG implementation in PyTorch
Two agents control rackets to bounce a ball over a net.
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. 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.
Agent's brain was a Deep Deterministic Policy Gradient, an Actor-Critic class Reinforcement Learning algorithm, implemented according to https://arxiv.org/abs/1509.02971 with Ornstein-Uhlenbeck random process.
Agent was able to solve the environment in 791 episodes (mean window starting at 791st episode exceeded solution threshold)
Packages:
- Pytorch
- Numpy
- UnityAgents
Additionaly, Jupyter notebook (or Jupyter lab) for displaying solution
You need to manually download environment binaries for your system:
- Linux: click here
- Mac OSX: click here
- Windows (32-bit): click here
- Windows (64-bit): click here
After downloading, unpack all files and place them in directory Tennis_Windows_x86_64/. Then, swap string in cell 2
env = UnityEnvironment(file_name="Tennis_Windows_x86_64/Tennis.exe")
with one proper to your OS. Notebook using pytorch implementation is in location Tennis.ipynb