FlagEmbedding icon indicating copy to clipboard operation
FlagEmbedding copied to clipboard

Adding validation set evaluation for reranker finetuning

Open jackkwok opened this issue 1 year ago • 1 comments

To avoid overfitting, I would like to add validation (eval) dataset to the finetuning. I see the BiTrainer is a subclass of Huggingface's Trainer class which accepts a eval_dataset argument. I plan to add the eval dataset within the initialization of BiTrainer (line).

Let me know if there are other considerations.

jackkwok avatar May 29 '24 20:05 jackkwok

@jackkwok Did you successfully add the validation dataset?If successful, can you share how to achieve it?

EEEnterURname avatar Jul 14 '24 06:07 EEEnterURname