code: def after_epoch(self, **_) -> None:
          
          if not self._total_batch_count_saved:
              self._total_batch_count = self._current_batch_count.copy()
              self._total_batch_count_saved = True
          self._current_batch_count.clear()
          self._current_stream_start = None
          self._current_stream_name = None
          erase_line()
docs: Reset progress counters. Save ``total_batch_count`` after the 1st epoch.