I posted my answer here: Is there a way to add metric to progress bar without calling `log`? - #2 by awaelchli