Installation and downloading of existing models, refer to link.
Need to install KenLM for as the language modeling prior:
pip install https://github.com/kpu/kenlm/archive/master.zipThe RSAModel is added as a wrapper around original code.
A list of files/models that we modify:
models/CaptionModel- We added RSA-based sampling methods in there