          tensorboard.on_train_begin(0)
          logger.verbose("Enabled TensorBoard Logging")
          return tensorboard
  
      def train_one_step(self, viewer, timelapse_kwargs):
          """ Running training on a batch of images for each side.
  
          Triggered from the training cycle in :class:`scripts.train.Train`.
 
</s>
    def toggle_mask(self):
        logger.info("Flipping mask display")
        self._samples.toggle_mask_display()

          tensorboard.on_train_begin(0)
          logger.verbose("Enabled TensorBoard Logging")
          return tensorboard
  
    def toggle_mask(self):
        logger.info("Flipping mask display")
        self._samples.toggle_mask_display()

      def train_one_step(self, viewer, timelapse_kwargs):
          """ Running training on a batch of images for each side.
  
          Triggered from the training cycle in :class:`scripts.train.Train`.
keep keep keep add keep keep keep keep
 - added push t to disable mask display at preview window

trsprmvd

- added extra checks for mask presence in the set
plugins/train/trainer/_base.py