huggingface / setfit

Efficient few-shot learning with Sentence Transformers
https://hf.co/docs/setfit
Apache License 2.0
2.24k stars 223 forks source link

Refactor training logs & warmup_proportion #475

Closed tomaarsen closed 10 months ago

tomaarsen commented 10 months ago

Resolves #473

Hello!

Pull Request overview

Details

Training logs

Now we display the number of unique pairs (instead of number of examples, which was broken). The rest is only updated slightly.

Update how warmup_steps are defined

It now uses number of training samples if training samples > args.max_steps alongside the warmup_proportion. This should make the warmup_proportion match reality.

Thanks @michael-brunzel for raising this!

HuggingFaceDocBuilderDev commented 10 months ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.