Skip to content
/ ToM Public

Code accompanying ICML 2021 paper "Few-shot Language Coordination by Modeling Theory of Mind"

License

Notifications You must be signed in to change notification settings

CLAW-Lab/ToM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

Code accompanying ICML 2021 paper "Few-shot Language Coordination by Modeling Theory of Mind"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages