Method update

omnivoice/training/checkpoint.py:66–83

Called every step to update the progress bar UI.

(
        self, step: int, loss: Optional[float] = None, lr: Optional[float] = None
    )

            )

    def update(
        self, step: int, loss: Optional[float] = None, lr: Optional[float] = None
    ):
        """
        Called every step to update the progress bar UI.
        """
        if self.progress_bar:
            self.progress_bar.update(1)

            # Update real-time metrics on the progress bar itself
            postfix = {}
            if loss is not None:
                postfix["loss"] = f"{loss:.4f}"
            if lr is not None:
                postfix["lr"] = f"{lr:.2e}"

            if postfix:
                self.progress_bar.set_postfix(postfix)

    def log_metrics(self, step: int, metrics: Dict[str, Any]):
        """