README.md

PyTorch implementation for the EACL 2021 paper

Modeling Coreference Relations in Visual Dialog

• bert_base folder: implementation of baseline model
• bert_pos folder: implementation of baseline model + pos constraint
• bert_sen folder: implementation of baseline model + nearest preference constraint
• bert_pos_senemd folder: implementation of the best model (baseline model + two constraints)

Set up dependencies

pip install -r dependencies.txt

Running code

Set up data path and parameters properly

python {bert_base/bert_pos/bert_sen/bert_pos_senemd}{train/evaluate}.py

Results on the online leaderboard