Code accompanying ICML 2021 paper "Few-shot Language Coordination by Modeling Theory of Mind"
The core code used in preparing data and training MAML speaker is ibr\_game/maml_speaker.py.
Stay tuned for the data and pretrained listeners.
ibr_game/beam_search.py is a lightweight batchified standalone beam search toolkit we designed during doing experiments.