search

UKPLab / sentence-transformers

State-of-the-Art Text Embeddings
https://www.sbert.net
Apache License 2.0
15.17k stars 2.47k forks source link

Epoch count for training_nli.py #370

Closed nandinib1999 closed 4 years ago

nandinib1999 commented 4 years ago

Hi,

I was working with training_nli.py and I saw that by default, the number of epochs is set to 1. I want to train the model for the task of semantic similarity.

How many training epochs do you recommend?

nreimers commented 4 years ago

In my experience 3-4 epochs are a good value for semantic similarity.

nandinib1999 commented 4 years ago

Thanks for your quick reply.

I am also planning to continue the training with sts benchmark dataset to train nli-stsb model. How many epochs are suitable for training_stsbenchmark_continue_training.py?

nreimers commented 4 years ago

Just try it out and you will see what works best

nandinib1999 commented 4 years ago

Okay, thanks.