  
      def train_one_step(self, viewer, timelapse_kwargs):
          """ Train a batch """
          logger.trace("Training one step: (iteration: %s)", self.model.iterations)
          do_preview = False if viewer is None else True
          do_timelapse = False if timelapse_kwargs is None else True
          loss = dict()
          for side, batcher in self.batchers.items():
              if self.pingpong.active and side != self.pingpong.side:
                  continue
              loss[side] = batcher.train_one_batch(do_preview)
 
</s> remove         self.target = batch[1] if do_preview else None
 </s> add         self.generate_preview(do_preview) </s> remove         self.samples = batch[0] if do_preview else None
 </s> add  </s> remove         q_name = "timelapse_{}".format(side) if is_timelapse else "train_{}".format(side)
 </s> add         q_name = "_{}".format(side)
        if is_preview:
            q_name = "{}{}".format("preview", q_name)
        elif is_timelapse:
            q_name = "{}{}".format("timelapse", q_name)
        else:
            q_name = "{}{}".format("train", q_name)
        do_preview = viewer is not None
        do_timelapse = timelapse_kwargs is not None
  
      def train_one_step(self, viewer, timelapse_kwargs):
          """ Train a batch """
          logger.trace("Training one step: (iteration: %s)", self.model.iterations)
        do_preview = viewer is not None
        do_timelapse = timelapse_kwargs is not None
        do_preview = viewer is not None
        do_timelapse = timelapse_kwargs is not None
          loss = dict()
          for side, batcher in self.batchers.items():
              if self.pingpong.active and side != self.pingpong.side:
                  continue
              loss[side] = batcher.train_one_batch(do_preview)
keep keep keep keep replace 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