egg/zoo/compo_vs_generalization/train.py
egg/zoo/language_bottleneck/guess_number/train.py