Imitation: Clean implementations of Imitation Learning algorithms

imitation is available on GitHub at

Main Features

  • Built on and compatible with Stable Baselines 3 (SB3).

  • Modular Pytorch implementations of Behavioral Cloning, DAgger, GAIL, and AIRL that can train arbitrary SB3 policies.

  • GAIL and AIRL have customizable reward and discriminator networks.

  • Scripts to train policies using SB3 and save rollouts from these policies as synthetic “expert” demonstrations.

  • Data structures and scripts for loading and storing expert demonstrations.