BRIO icon indicating copy to clipboard operation
BRIO copied to clipboard

Is it possible to train Huggingface model using BRIO class?

Open dleunji opened this issue 2 years ago • 1 comments

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.

dleunji avatar Aug 20 '22 09:08 dleunji

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.

yixinL7 avatar Sep 06 '22 15:09 yixinL7