ToM

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.

Misc

ibr_game/beam_search.py is a lightweight batchified standalone beam search toolkit we designed during doing experiments.

Acknowledgement

We used and adapted code from MAML++ and S2P.