          logger.debug("Loading Trainer")
          trainer = PluginLoader.get_trainer(model.trainer)
          trainer = trainer(model,
                            self.images,
                            self.args.batch_size)
          logger.debug("Loaded Trainer")
          return trainer
  
      def run_training_cycle(self, model, trainer):
          """ Perform the training cycle """
 
</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     def __init__(self, model, use_mask, coverage_ratio, batchers):
 </s> add     def __init__(self, model, use_mask, coverage_ratio, preview_images, batchers): </s> remove     def __init__(self, model, images, batch_size):
 </s> add     def __init__(self, model, images, batch_size, configfile): </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
                          self.args.batch_size,
                          self.args.configfile)
          logger.debug("Loading Trainer")
          trainer = PluginLoader.get_trainer(model.trainer)
          trainer = trainer(model,
                            self.images,
                          self.args.batch_size,
                          self.args.configfile)
          logger.debug("Loaded Trainer")
          return trainer
  
      def run_training_cycle(self, model, trainer):
          """ Perform the training cycle """
keep keep keep keep replace keep keep keep keep keep
 Expose Augmentation Options to config 
scripts/train.py