Robust Imitation of Diverse Behaviors