          self.use_mask = use_mask
          self.side = side
          self.images = images
          self.target = None
          self.samples = None
          self.mask = None
  
          self.feed = self.load_generator().minibatch_ab(images, batch_size, self.side)
          self.preview_feed = None
 
</s> remove     def __init__(self, side, images, model, use_mask, batch_size):
        logger.debug("Initializing %s: side: '%s', num_images: %s, batch_size: %s)",
                     self.__class__.__name__, side, len(images), batch_size)
 </s> add     def __init__(self, side, images, model, use_mask, batch_size, config):
        logger.debug("Initializing %s: side: '%s', num_images: %s, batch_size: %s, config: %s)",
                     self.__class__.__name__, side, len(images), batch_size, config) </s> remove         num_images = self.model.training_opts.get("preview_images", 14)
 </s> add         num_images = self.config.get("preview_images", 14) </s> remove                      "batchers: '%s')", self.__class__.__name__, model, use_mask,
                     coverage_ratio, batchers)
 </s> add                      "preview_images: %s, batchers: '%s')", self.__class__.__name__, model,
                     use_mask, coverage_ratio, preview_images, batchers)
        self.preview_images = preview_images </s> remove                                        batch_size)
 </s> add                                        batch_size,
                                       self.config) </s> remove                                             training_opts.get("coverage_ratio", 0.625))
 </s> add                                             training_opts.get("coverage_ratio", 0.625),
                                            config)
        self.config = config
          self.use_mask = use_mask
          self.side = side
          self.images = images
        self.config = config
          self.target = None
          self.samples = None
          self.mask = None
  
          self.feed = self.load_generator().minibatch_ab(images, batch_size, self.side)
          self.preview_feed = None
keep keep add keep keep keep keep keep keep
 Expose Augmentation Options to config 
plugins/train/trainer/_base.py