for name, label in self._ENV_METRICS:
if name in env_metrics:
postfix[label] = self._format_metric(name, env_metrics[name])
if "step_time" in postfix:
postfix = {"step_time": postfix.pop("step_time"), **postfix}
return postfix
def on_train_begin(self, state: TrainerState, **kwargs: Any) -> None:
"""Create a single full-training progress bar on global rank zero.