                                 l_inf_norm=losses.LInfNorm(),
                                 ssim=losses.DSSIMObjective(),
                                 gmsd=losses.GMSDLoss(),
                                 pixel_gradient_diff=losses.GradientLoss())
          self._uses_l2_reg = ["ssim", "gmsd"]
          self._inputs = None
          self._names = []
          self._funcs = {}
          logger.debug("Initialized: %s", self.__class__.__name__)
  
 
</s> remove                  "penalty will only be applied if SSIM or GMSD is selected for the main loss "
                 "function, otherwise it is ignored.\n\nThe value given here is as a percentage "
                 "weight of the main loss function. For example:"
 </s> add                  "penalty will only be applied if SSIM, MS-SSIM or GMSD is selected for the main "
                 "loss function, otherwise it is ignored."
                 "\n\nThe value given here is as a percentage weight of the main loss function. "
                 "For example:" </s> remove             choices=["mae", "mse", "logcosh", "smooth_loss", "l_inf_norm", "ssim", "gmsd",
                     "pixel_gradient_diff"],
 </s> add             choices=["mae", "mse", "logcosh", "smooth_loss", "l_inf_norm", "ssim", "ms_ssim",
                     "gmsd", "pixel_gradient_diff"],
        self._uses_l2_reg = ["ssim", "ms_ssim", "gmsd"]
                                 l_inf_norm=losses.LInfNorm(),
                                 ssim=losses.DSSIMObjective(),
                                 gmsd=losses.GMSDLoss(),
                                 pixel_gradient_diff=losses.GradientLoss())
        self._uses_l2_reg = ["ssim", "ms_ssim", "gmsd"]
          self._inputs = None
          self._names = []
          self._funcs = {}
          logger.debug("Initialized: %s", self.__class__.__name__)
  
keep keep keep keep replace keep keep keep keep keep
 training - Add MS-SSIM loss function 
plugins/train/model/_base.py