              section=section, title="penalized_mask_loss", datatype=bool, default=True,
              info="If using a mask, This penalizes the loss for the masked area, to give higher "
                   "priority to the face area. \nShould increase overall quality and speed up "
                   "training. This should probably be left at True")
          self.add_item(
              section=section, title="preview_images", datatype=int, default=14, min_max=(2, 16),
              rounding=2, fixed=False,
              info="Number of sample faces to display for each side in the preview when training.")
          logger.debug("Set global config")
  
      def load_module(self, filename, module_path, plugin_type):
          """ Load the defaults module and add defaults """
          logger.debug("Adding defaults: (filename: %s, module_path: %s, plugin_type: %s",
 
</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  </s> remove         logger.debug("Initializing %s: (input_size: %s, output_size: %s, coverage_ratio: %s)",
                     self.__class__.__name__, input_size, output_size, coverage_ratio)
        # Transform args
        self.rotation_range = 10  # Range to randomly rotate the image by
        self.zoom_range = 0.05  # Range to randomly zoom the image by
        self.shift_range = 0.05  # Range to randomly translate the image by
        self.random_flip = 0.5  # Chance to flip the image horizontally
 </s> add         logger.debug("Initializing %s: (input_size: %s, output_size: %s, coverage_ratio: %s, "
                     "config: %s)", self.__class__.__name__, input_size, output_size,
                     coverage_ratio, config)
        self.config = config </s> remove     def __init__(self, input_size, output_size, coverage_ratio):
 </s> add     def __init__(self, input_size, output_size, coverage_ratio, config): </s> remove         augment_color:      Perform random shifting of LAB colors
 </s> add         augment_color:      Perform random shifting of L*a*b* colors </s> remove         num_images = self.model.training_opts.get("preview_images", 14)
 </s> add         num_images = self.config.get("preview_images", 14)
              section=section, title="penalized_mask_loss", datatype=bool, default=True,
              info="If using a mask, This penalizes the loss for the masked area, to give higher "
                   "priority to the face area. \nShould increase overall quality and speed up "
                   "training. This should probably be left at True")




          logger.debug("Set global config")
  
      def load_module(self, filename, module_path, plugin_type):
          """ Load the defaults module and add defaults """
          logger.debug("Adding defaults: (filename: %s, module_path: %s, plugin_type: %s",
keep keep keep keep replace replace replace replace keep keep keep keep keep
 Expose Augmentation Options to config 
plugins/train/_config.py