Reorder AIRLTrainer methods
Reorder AIRLTrainer methods

Seems easier to read if I move all the private methods to the
bottom.

HumanCompatibleAI/imitation,HumanCompatibleAI/imitation,humancompatibleai/imitation,humancompatibleai/imitation