          self.samples = batch[0] if do_preview else None
          batch = batch[1:]   # Remove full size samples from batch
          if self.use_mask:
              batch = self.compile_mask(batch)
          self.target = batch[1] if do_preview else None
          return batch
  
      def compile_mask(self, batch):
          """ Compile the mask into training data """
          logger.trace("Compiling Mask: (side: '%s')", self.side)
 
</s> remove         self.samples = batch[0] if do_preview else None
 </s> add  </s> remove         do_preview = False if viewer is None else True
        do_timelapse = False if timelapse_kwargs is None else True
 </s> add         do_preview = viewer is not None
        do_timelapse = timelapse_kwargs is not None </s> remove             logger.trace("Putting to batch queue: (side: '%s', is_timelapse: %s)",
                         side, is_timelapse)
 </s> add             logger.trace("Putting to batch queue: (side: '%s', is_display: %s)",
                         side, is_display)
        self.generate_preview(do_preview)
          self.samples = batch[0] if do_preview else None
          batch = batch[1:]   # Remove full size samples from batch
          if self.use_mask:
              batch = self.compile_mask(batch)
        self.generate_preview(do_preview)
          return batch
  
      def compile_mask(self, batch):
          """ Compile the mask into training data """
          logger.trace("Compiling Mask: (side: '%s')", self.side)
keep keep keep keep replace keep keep keep keep keep
 Color Augmentation method (#752)

* Color Augmentation: Implement testing code

* Tweak Contrast and Lighting augmentation amounts

* Prevent color augmentation from entering timelapse

* Remove all augmentations from preview images 
plugins/train/trainer/_base.py