              title="loss_function",
              datatype=str,
              group="loss",
              default="ssim",
              choices=["mae", "mse", "logcosh", "smooth_loss", "l_inf_norm", "ssim", "gmsd",
                       "pixel_gradient_diff"],
              info="The loss function to use."
                   "\n\t MAE - Mean absolute error will guide reconstructions of each pixel "
                   "towards its median value in the training dataset. Robust to outliers but as "
                   "a median, it can potentially ignore some infrequent image types in the dataset."
                   "\n\t MSE - Mean squared error will guide reconstructions of each pixel "
 
</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         self._uses_l2_reg = ["ssim", "gmsd"]
 </s> add         self._uses_l2_reg = ["ssim", "ms_ssim", "gmsd"]
            choices=["mae", "mse", "logcosh", "smooth_loss", "l_inf_norm", "ssim", "ms_ssim",
                     "gmsd", "pixel_gradient_diff"],
              title="loss_function",
              datatype=str,
              group="loss",
              default="ssim",
            choices=["mae", "mse", "logcosh", "smooth_loss", "l_inf_norm", "ssim", "ms_ssim",
                     "gmsd", "pixel_gradient_diff"],
            choices=["mae", "mse", "logcosh", "smooth_loss", "l_inf_norm", "ssim", "ms_ssim",
                     "gmsd", "pixel_gradient_diff"],
              info="The loss function to use."
                   "\n\t MAE - Mean absolute error will guide reconstructions of each pixel "
                   "towards its median value in the training dataset. Robust to outliers but as "
                   "a median, it can potentially ignore some infrequent image types in the dataset."
                   "\n\t MSE - Mean squared error will guide reconstructions of each pixel "
keep keep keep keep replace replace keep keep keep keep keep
 training - Add MS-SSIM loss function 
plugins/train/_config.py