code: def on_batch_end(self, iteration:int, **kwargs)->None:
          "Callback function that writes batch end appropriate data to Tensorboard."
          super().on_batch_end(iteration=iteration, **kwargs)
          if iteration == 0: return
          if iteration % self.visual_iters == 0: self._write_images(iteration=iterationf on_batch_end(self, iteration:int, **kwargs)->None:
          "Callback function that writes batch end appropriate data to Tensorboard."
          super().on_batch_end(iteration=iteration, **kwargs)
          if iteration == 0: return
          if iteration % self.visual_iters == 0: self._write_images(iteration=iteration)
docs: Callback function that writes batch end appropriate data to Tensorboard.