Imitation: Clean implementations of Imitation Learning algorithms
imitation is available on GitHub at http://github.com/HumanCompatibleAI/imitation
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.