Progress bar metrics

[stalevna]

Hi everyone!
I want to learn how to change progress bar metrics when training. I used to have loss showing in the bar, but now I have train_step_timing which is not very useful. Could you help me figure out where I can change these settings? I tried to search in the Pytorch files, but the places where you change it were commented out.
This is the example of how it shows right now:
32:29<5:19:08, 4.81it/s, v_num=0-12, train_step_timing in s=0.122

[titu1994]

It depends on the model, but in general were logging the results in a models training_step() and validation_step()

[stalevna]

it is an ASR model. I have checked and the training_step() returns loss and that is exactly what I need but for some reason in the progress bar I still get train_step_timing. I will try to install and put RichProgressBar in callbacks -> trainer = Trainer(callbacks=RichProgressBar()) and see if it helps

[stalevna]

Unfortunately, RichProgressBar didn't help

Going to look into pytorch_lightning, maybe logging results are overwritten somewhere