Skip to content

Proximal Policy Optimization using Pytorch and the Unity Reacher environment.

License

Notifications You must be signed in to change notification settings

primeMover2011/PromixalPolicyOptimization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Continuous Control using Proximal Policy Optimization

Introduction

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

Trained Agent

In this environment, a double-jointed arm can move to target locations. A reward of +0.1 is provided for each step that the agent's hand is in the goal location. Thus, the goal of your agent is to maintain its position at the target location for as many time steps as possible.

The observation space consists of 33 variables corresponding to position, rotation, velocity, and angular velocities of the arm.

Each action is a vector with four numbers, corresponding to torque applicable to two joints. Every entry in the action vector is a number between -1 and 1.

Project Details

This version of the Unity environment contains 20 identical agents. The agents are trained using PPO to get an average score of +30 (over 100 consecutive episodes, and over all 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 20 (potentially different) scores. We then take the average of these 20 scores.
  • This yields an average score for each episode (where the average is over all 20 agents).

When the average of the average scores is at least +30, the environment is considered solved.

Getting Started

  1. Clone this repository, and navigate to the PromixalPolicyOptimization/ folder. Then, install several dependencies.
git clone https://github.com/primeMover2011/PromixalPolicyOptimization.git
cd PromixalPolicyOptimization
  1. 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.)

  2. Extract the contents of the file to a folder of you choice, preferably as a subfolder of this repository.

  3. Install conda

  4. cd in to the directory where you cloned this repository, create a virtual environment and install the required python packages using these commands

cd PromixalPolicyOptimization
conda env create -f environment.yml

activate the environment using

conda activate PromixalPolicyOptimization

update the location of the environment in main.py and in test.py

    env = UnityEnvironment(file_name='YOUR_PATH_HERE', base_port=64739)
  1. Watch the pretrained agent.
  python test.py
  1. Train your own agent using default parameters.
  python main.py
  1. Read [the report] and play around with the code. Best starting point is defining your own experiments using some changes in hyper parameters.

in main.py change

    scores = [

    run_experiment(hidden_size=256, lr=1e-3, max_episodes=500, mini_batch_size=128,
                   nrmlz_adv=True, num_steps=2048, ppo_epochs=4, threshold_reward=30,
                   gamma=0.99, tau=0.95, clip_gradients=True)
    ]

Enjoy!

About

Proximal Policy Optimization using Pytorch and the Unity Reacher environment.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages