Use amp / FP16

Open PhilipMay opened this issue 3 years ago • 2 comments

Hi, since the sentence transformers support FP16 training (by setting use_amp=True) can this also be build in here?

Not sure if this is a FR or a question. :-)

Oct 28 '22 15:10 PhilipMay

Should be easy to add it here:

https://github.com/huggingface/setfit/blob/7735e8e3b208edb8dfb549beb16e585453c5f44e/src/setfit/trainer.py#L276

Oct 29 '22 14:10 PhilipMay

see PR #134

Oct 30 '22 11:10 PhilipMay

closing this because PR is merged

Oct 31 '22 18:10 PhilipMay