Variation Inverse Reinforcement Learning

Implementation of Adversarial Imitation Via Variational Inverse Reinforcement Learning.

The code is an adaption of inverse-rl repository that contains the implementations of state-of-the-art imitation & inverse reinforcement learning algorithms.

Requirements

• Rllab
  • Use our base.py by replacing from rllab.sampler.base import BaseSampler to from base import BaseSampler in the file sandbox/rocky/tf/samplers/vectorized_sampler.py
• TensorFlow

Examples

Running the Ant gym environment

1. Collect expert data

   python ant_data_collect.py

2. Run Inverse Reinforcement Learning:

   python ant_irl.py

3. Run transfer learning on disabled-ant

   python ant_transfer_disabled.py