Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Add option to disable the progress bar

See original GitHub issue

Hi,

unless I missed something, it is currently not possible to disable the progress bar when training a SetFit model using SetFitTrainer. I suggest adding an option for this.

Motivation: I am using SetFit in a Jupyter Notebook (not sure why it seems to print more than in the SetFit example notebook) and the output makes it rather ugly. Moreover, one might want not to display progress bars regardless of the notebook setting.

Problematic calls: https://github.com/huggingface/setfit/blob/fa1021d2355f0cb3a2c85732ee7ffe44b0cef0d1/src/setfit/trainer.py#L374 https://github.com/huggingface/setfit/blob/fa1021d2355f0cb3a2c85732ee7ffe44b0cef0d1/src/setfit/trainer.py#L388

Issue Analytics

State:
Created 10 months ago
Comments:5

Top GitHub Comments

1reaction

tomaarsencommented, Dec 2, 2022

Agreed. I’ve made a PR to handle the SetFit side of this issue, i.e. with show_progress_bar.

0reactions

kgourgoucommented, Dec 2, 2022

I get the pretty progress bars with the current version of SetFit, all I need to do is make sure ipywidgets is up-to-date and I use from tqdm.auto import tqdm.

However, not being able to turn off all progress bars is an issue.