transformers stop the printing, while still logging to wandb, tensorboard, csv etc

hf_tf_no_print_progress.py

from transformers.trainer_callback import ProgressCallback
class ProgressCallbackNoPrint(ProgressCallback):
    """ProgressCallback that doesn't print anything
    
    Usage: 
        # at top of file, after transformers import, before settin up trainer, monkey patch the default progress callback
        import transformers
        transformers.DEFAULT_PROGRESS_CALLBACK = ProgressCallbackNoPrint

        # OR you can do it this way

        trainer = Trainer(
            ..., 
            logging_steps=1,
            callbacks=[ProgressCallbackNoPrint()]
        )
        rm_old_prog_cb(trainer)
        trainer.train()
                
    @url: https://gist.github.com/wassname/c3e1d00b21e9fa6d6bce96ff7263eafb
    """
    def on_log(self, *args, **kwargs):
        pass
    
def rm_old_prog_cb(trainer):
    for cb in trainer.callback_handler.callbacks:
        if isinstance(cb, ProgressCallback):
            if not isinstance(cb, ProgressCallbackNoPrint):
                trainer.remove_callback(cb)