              info="If using a mask, Use DSSIM loss for Mask training rather than Mean Absolute "
                   "Error\nMay increase overall quality.")
          self.add_item(
              section=section, title="penalized_mask_loss", datatype=bool, default=True,
              info="If using a mask, Use Penalized loss for Mask training. Can stack with DSSIM.\n"
                   "May increase overall quality.")
  
          # << DFAKER OPTIONS >> #
          section = "model.dfaker"
          self.add_section(title=section,
                           info="Dfaker Model (Adapted from https://github.com/dfaker/df)" +
 
</s> remove             info="Use reflect padding rather than zero padding.")
        self.add_item(
            section=section, title="dssim_mask_loss", datatype=bool, default=True,
            info="If using a mask, Use DSSIM loss for Mask training rather than Mean Absolute "
                 "Error\nMay increase overall quality.")
 </s> add             info="Use reflect padding rather than zero padding. Only enable this option if the "
                 "model you are training has a distinct line appearing around the edge of the "
                 "swap area.") </s> remove         if self.config.get("dssim_mask_loss", False):
            if side == "a" and not self.predict and initialize:
                logger.verbose("Using DSSIM Loss for mask")
            mask_loss_func = DSSIMObjective()
        else:
            if side == "a" and not self.predict:
                logger.verbose("Using Mean Absolute Error Loss for mask")
            mask_loss_func = losses.mean_absolute_error
 </s> add         if side == "a" and not self.predict:
            logger.verbose("Using Mean Absolute Error Loss for mask")
        mask_loss_func = losses.mean_absolute_error </s> remove         optimizer = self.get_optimizer(lr=5e-5, beta_1=0.5, beta_2=0.999)
 </s> add         learning_rate = self.config.get("learning_rate", 5e-5)
        optimizer = self.get_optimizer(lr=learning_rate, beta_1=0.5, beta_2=0.999)
            info="If using a mask, Use Penalized loss for Mask training. \n"
                 "May increase overall quality. This should probably be left at True")
              info="If using a mask, Use DSSIM loss for Mask training rather than Mean Absolute "
                   "Error\nMay increase overall quality.")
          self.add_item(
              section=section, title="penalized_mask_loss", datatype=bool, default=True,
            info="If using a mask, Use Penalized loss for Mask training. \n"
                 "May increase overall quality. This should probably be left at True")
            info="If using a mask, Use Penalized loss for Mask training. \n"
                 "May increase overall quality. This should probably be left at True")
  
          # << DFAKER OPTIONS >> #
          section = "model.dfaker"
          self.add_section(title=section,
                           info="Dfaker Model (Adapted from https://github.com/dfaker/df)" +
keep keep keep keep replace replace keep keep keep keep keep
 Correct dfaker config.
Make lr configurable 
plugins/train/_config.py