AMP-GAIL implementation

In the usual GAIL implementation, the generator step is carried out a few (2-3) times for every discriminator step. I cannot find this in the AMP-GAIL implementation (learning/amp_continuous.py). It looks like it executes a single generator and discriminator step.