search

ianporada / coref-reeval

A controlled reevaluation of coreference resolution models
5 stars 1 forks source link

Train and Inference scripts for the LingMess implementation? #4

Open sinamps opened 5 months ago

sinamps commented 5 months ago

Hello, I was wondering if a training and inference script is released for the LingMess implementation or the same one from here: https://github.com/ianporada/coref-reeval/blob/main/models/encoder_based/WLC_S2E_C2F/src/train.py should be adapted/used?

another question is do I need a separate data module for LingMess?

Thank you for the great work of organizing and standardizing multiple models and datasets for coreference.

ianporada commented 5 months ago

I believe LingMess should work with the same trainer. I'll try to run it to make sure. (@XiyuanZou implemented the LingMess model and is planning to clean up the code when he returns from NAACL).

ianporada commented 5 months ago

LingMess does indeed use a different train script. I've added the script at models/encoder_based/experiments/train_lingmess_teacher/train.py as well as a little bit more documentation.