search

yixinL7 / BRIO

ACL 2022: BRIO: Bringing Order to Abstractive Summarization
330 stars 43 forks source link

Is it possible to train Huggingface model using BRIO class? #15

Open dleunji opened 2 years ago

dleunji commented 2 years ago

Hi, your codes and paper are really interested! In README, generation with hugggingface is mentioned, but training was unmentioned.

Is it possible to train huggingface model using BRIO class?

Thank you.

yixinL7 commented 2 years ago

Hi, thank you for your interest in our work! You may finetune the BRIO Huggingface model using MLE. But to train the model using our method you will need to use the model class defined in this repo.