          self.batchers = {side: Batcher(side,
                                         images[side],
                                         self.model,
                                         self.use_mask,
                                         batch_size)
                           for side in self.sides}
  
          self.tensorboard = self.set_tensorboard()
          self.samples = Samples(self.model,
                                 self.use_mask,
 
</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     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)
                                       batch_size,
                                       self.config)
          self.batchers = {side: Batcher(side,
                                         images[side],
                                         self.model,
                                         self.use_mask,
                                       batch_size,
                                       self.config)
                           for side in self.sides}
  
          self.tensorboard = self.set_tensorboard()
          self.samples = Samples(self.model,
                                 self.use_mask,
keep keep keep keep replace keep keep keep keep keep
 Expose Augmentation Options to config 
plugins/train/trainer/_base.py