  
      def set_preview_feed(self):
          """ Set the preview dictionary """
          logger.debug("Setting preview feed: (side: '%s')", self.side)
          batchsize = min(len(self.images), self.model.training_opts.get("preview_images", 14))
          self.preview_feed = self.load_generator().minibatch_ab(self.images,
                                                                 batchsize,
                                                                 self.side,
                                                                 do_shuffle=True,
                                                                 is_preview=True)
 
</s> remove         num_images = self.model.training_opts.get("preview_images", 14)
 </s> add         num_images = self.config.get("preview_images", 14) </s> remove                         self.model.training_opts.get("preview_images", 14))
 </s> add                         self.preview_images) </s> remove         generator = TrainingDataGenerator(input_size, output_size, self.model.training_opts)
 </s> add         generator = TrainingDataGenerator(input_size,
                                          output_size,
                                          self.model.training_opts,
                                          self.config) </s> remove         preview_images = self.config.get("preview_images", 14)
        preview_images = min(max(preview_images, 2), 16)
        self.training_opts["preview_images"] = preview_images
 </s> add 
        preview_images = self.config.get("preview_images", 14)
        preview_images = min(max(preview_images, 2), 16)
        batchsize = min(len(self.images), preview_images)
  
      def set_preview_feed(self):
          """ Set the preview dictionary """
          logger.debug("Setting preview feed: (side: '%s')", self.side)
        preview_images = self.config.get("preview_images", 14)
        preview_images = min(max(preview_images, 2), 16)
        batchsize = min(len(self.images), preview_images)
          self.preview_feed = self.load_generator().minibatch_ab(self.images,
                                                                 batchsize,
                                                                 self.side,
                                                                 do_shuffle=True,
                                                                 is_preview=True)
keep keep keep keep replace keep keep keep keep keep
 Expose Augmentation Options to config 
plugins/train/trainer/_base.py