  
  class TrainerBase():
      """ Base Trainer """
  
      def __init__(self, model, images, batch_size):
          logger.debug("Initializing %s: (model: '%s', batch_size: %s)",
                       self.__class__.__name__, model, batch_size)
          self.batch_size = batch_size
          self.model = model
          self.model.state.add_session_batchsize(batch_size)
 
</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     def __init__(self, model, use_mask, coverage_ratio, batchers):
 </s> add     def __init__(self, model, use_mask, coverage_ratio, preview_images, batchers): </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
    def __init__(self, model, images, batch_size, configfile):
  
  class TrainerBase():
      """ Base Trainer """
  
    def __init__(self, model, images, batch_size, configfile):
          logger.debug("Initializing %s: (model: '%s', batch_size: %s)",
                       self.__class__.__name__, model, batch_size)
          self.batch_size = batch_size
          self.model = model
          self.model.state.add_session_batchsize(batch_size)
keep keep keep keep replace keep keep keep keep keep
 Expose Augmentation Options to config 
plugins/train/trainer/_base.py