                                     self.config["mixed_precision"],
                                     self.config["allow_growth"],
                                     self._is_predict)
  
          logger.debug("Initialized ModelBase (%s)", self.__class__.__name__)
  
      @property
      def model(self):
          """:class:`Keras.models.Model`: The compiled model for this plugin. """
 
</s> remove     @property
    def _selected_mask_loss(self):
        """ :func:`keras.losses.Loss`: The selected mask loss function. Currently returns mean
        standard error as the default function. """
        loss_func = self._loss_dict["mse"]
        logger.debug("loss_func: %s", loss_func)
        return loss_func

 </s> add  </s> remove         self._inputs = inputs
        self._names = self._get_loss_names(outputs)
        self._funcs = self._get_loss_functions()
        self._names.insert(0, "total")
 </s> add         self._inputs = None
        self._names = None
        self._funcs = None </s> remove     """ Holds loss names and functions for an Autoencoder.

    Parameters
    ----------
    inputs: list
        A list of input tensors to the model in the order ("a", "b")
    outputs: list
        A list of output tensors to the model in the order ("a", "b")
    """
    def __init__(self, inputs, outputs):
        logger.debug("Initializing %s: (inputs: %s, outputs: %s)",
                     self.__class__.__name__, inputs, outputs)
 </s> add     """ Holds loss names and functions for an Autoencoder. """
    def __init__(self):
        logger.debug("Initializing %s", self.__class__.__name__) </s> remove             self._state.add_session_loss_names(loss.names)
 </s> add             self._state.add_session_loss_names(self._loss.names)
        self._loss = _Loss()
                                     self.config["mixed_precision"],
                                     self.config["allow_growth"],
                                     self._is_predict)
        self._loss = _Loss()
  
          logger.debug("Initialized ModelBase (%s)", self.__class__.__name__)
  
      @property
      def model(self):
          """:class:`Keras.models.Model`: The compiled model for this plugin. """
keep keep add keep keep keep keep keep keep
 Training Config - Split Loss to it's own section 
plugins/train/model/_base.py