Learning rate (lr) and training epochs (epoch) for pipeline training based on SBERT.
. | LR . | Bayes LR . | GP Reg . | |||
---|---|---|---|---|---|---|
lr . | epoch . | lr . | epoch . | lr . | epoch . | |
STS-B | 0.1 | 100 | 0.01 | 2500 | 0.1 | 25 |
EBMSASS | 0.1 | 15 | 0.01 | 10000 | 0.1 | 25 |
MedSTS | 0.1 | 100 | 0.01 | 8500 | 0.1 | 25 |
Yelp | 0.1 | 600 | 0.01 | 2500 | 0.1 | 25 |
en-zh | 0.1 | 50 | 0.03 | 300 | 0.1 | 200 |
ru-en | 0.1 | 40 | 0.03 | 400 | 0.1 | 200 |
si-en | 0.1 | 199 | 0.03 | 300 | 0.1 | 1000 |
. | LR . | Bayes LR . | GP Reg . | |||
---|---|---|---|---|---|---|
lr . | epoch . | lr . | epoch . | lr . | epoch . | |
STS-B | 0.1 | 100 | 0.01 | 2500 | 0.1 | 25 |
EBMSASS | 0.1 | 15 | 0.01 | 10000 | 0.1 | 25 |
MedSTS | 0.1 | 100 | 0.01 | 8500 | 0.1 | 25 |
Yelp | 0.1 | 600 | 0.01 | 2500 | 0.1 | 25 |
en-zh | 0.1 | 50 | 0.03 | 300 | 0.1 | 200 |
ru-en | 0.1 | 40 | 0.03 | 400 | 0.1 | 200 |
si-en | 0.1 | 199 | 0.03 | 300 | 0.1 | 1000 |