egg/zoo/compo_vs_generalization/train.py
egg/zoo/compositional_efficiency/discrete.py

egg/zoo/compo_vs_generalization/data.py
egg/zoo/compositional_efficiency/dataset.py

egg/zoo/compo_vs_generalization/train.py
egg/zoo/compositional_efficiency/continuous.py