search

Ravoxsg / SummaReranker

Source code for SummaReranker (ACL 2022)
MIT License
24 stars 10 forks source link

How to train the re-ranker model? #4

Closed ithieund closed 2 years ago

ithieund commented 2 years ago

Dear Mr. @Ravoxsg , In the readme, you only provide the pre-trained model checkpoints to reproduce the paper result. I couldn't find the section about how to train the re-ranker model to get that checkpoints. Could you please show me how to train the re-ranker model and save the pre-trained checkpoint file for next evaluation?

Thank you so much!

Ravoxsg commented 2 years ago

Hi @ithieund , I will release code for training soon (ETA ~ few weeks). So far I have released code for preparing the data + run the model with the 3 checkpoints I have shared.

ithieund commented 2 years ago

Hi @ithieund , I will release code for training soon (ETA ~ few weeks). So far I have released code for preparing the data + run the model with the 3 checkpoints I have shared.

Thank you. I'll wait for that.

ithieund commented 2 years ago

Hi @ithieund , I will release code for training soon (ETA ~ few weeks). So far I have released code for preparing the data + run the model with the 3 checkpoints I have shared.

Dear Mr. @Ravoxsg , It's about a month now so can you please release the training script next week? Thank you so much!

Ravoxsg commented 2 years ago

Hi @ithieund

I was busy with EMNLP deadline. I released the training pipeline! Please check it out.

Ravoxsg commented 2 years ago

By default everything is trained on Reddit-TIFU

ithieund commented 2 years ago

Hi @ithieund

I was busy with EMNLP deadline. I released the training pipeline! Please check it out.

Thank you so much, Mr. @Ravoxsg !