code: def increment(self, batch_size):
          
          self.example_count += batch_size
          self.example_total += batch_size
          if self.log_unit == "seconds":
              self.unit_count = int(self.timer.elapsed())
              self.unit_total = int(self.timer.total_elapsed())
          elif self.log_unit == "examples":
              self.unit_count = self.example_count
              self.unit_total = self.example_total
          elif self.log_unit == "batches":
              self.unit_count += 1
              self.unit_total += 1
          elif self.log_unit == "epochs":
              # Track epoch by example count because otherwise we only know when
              # a new epoch starts, not when an epoch ends
              if self.example_count >= self.epoch_size:
                  self.unit_count += 1
                  self.unit_total += 1
          else:
              raise Exception(f"Unrecognized log_unit: {self.log_unit}")
docs: Update the total and relative unit counts