Pytorch implementation for "End-to-End Differentiable Adversarial Imitation Learning", by Nir Baram, Oron Anschel, Itai Caspi, Shie Mannor.
Install the required dependencies using the following command:
pip install -r requirements.txt
The code is based on the tensorflow implementation: https://github.com/itaicaspi/mgail
Run the following command to train the Mujoco Hopper-v2 environment by imitating an expert trained with SAC or PPO:
python main.py