def add_pbar_task(self, name, total):
        """Adds a task to the progress bar.

        Args:
            name: Name of the new task.
            total: Total number of steps (samples) contained in the task.

        Returns:
            The task ID.
        """
        assert isinstance(self.pbar_kwargs, dict)
        self.pbar.append(tqdm(desc=name, total=total, **self.pbar_kwargs))
        return len(self.pbar) - 1
Adds a task to the progress bar.

        Args:
            name: Name of the new task.
            total: Total number of steps (samples) contained in the task.

        Returns:
            The task ID.
        
utils/logger.py