neroRL 0.0.4

A library for Deep Reinforcement Learning (PPO) in PyTorch

neroRL

neroRL is a PyTorch based research framework for Deep Reinforcement Learning specializing on Recurrent Proximal Policy Optimization. Its focus is set on environments that are procedurally generated, while providing some usefull tools for experimenting and analyzing a trained behavior.

Features

• Environments:
  • Obstacle Tower
  • Unity ML-Agents
  • Procgen
  • Gym-Minigrid (Vector (one-hot) or Visual Observations (84x84x3))
  • Gym CartPole using masked velocity
  • DM Ballet
• Proximal Policy Optimization (PPO)
  • Discrete and Multi-Discrete Action Spaces
  • Vector and Visual Observation Spaces (either alone or simultaneously)
  • Recurrent Policies using Truncated Backpropagation Through Time
  • Shared and None-Shared Parameters across the Policy and the Value Function
• Decoupled Proximal Policy Optimization
  • Same features as PPO, but the parameters, as well as the gradients, are decoupled
  • Decoupled Advantage Actor-Critic (DAAC, Raileanu & Fergus, 2021)
    • The actor network estimates the policy and the advantage function

Obstacle Tower Challenge

Originally, this work started out by achieving the 7th place during the Obstacle Tower Challenge by using a relatively simple FFCNN. This video presents some footage of the approach and the trained behavior:

Recently we published a paper at CoG 2020 (best paper candidate) that analyzes the taken approach. Additionally the model was trained on 3 level designs and was evaluated on the two left out ones. The results can be reproduced using the obstacle-tower-challenge branch.

Getting Started

To get started check out the docs!