  
      def _get_loss_functions(self):
          """ Set the loss functions.
  
          Returns
          -------
          list
              A list of loss functions to apply to the model
          """
          selected_loss = self._loss_dict[self._config["loss_function"]]
          mask_loss = self._loss_dict[self._config["mask_loss_function"]]
          loss_funcs = []
          for name in self._names:
 
</s> remove         mask_loss = self._loss_dict[self._config["mask_loss_function"]]
        loss_funcs = []
        for name in self._names:
 </s> add         for name, output_name in zip(self._names, output_names): </s> remove     def _get_loss_functions(self):
 </s> add     def _get_loss_functions(self, output_names): </s> remove 
        Returns
        -------
        list
            A list of names for the losses to be applied to the model
 </s> add  </s> remove         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")
 </s> add         model: :class:`keras.models.Model`
            The model that is to be trained </s> remove                 loss_funcs.append(mask_loss)
 </s> add                 loss_func = self._loss_dict[self._config["mask_loss_function"]]
        Adds the loss functions to the :attr:`functions` dictionary.

        Parameters
        ----------
        output_names: list
            The output names from the model
  
      def _get_loss_functions(self):
          """ Set the loss functions.
  
        Adds the loss functions to the :attr:`functions` dictionary.

        Parameters
        ----------
        output_names: list
            The output names from the model
        Adds the loss functions to the :attr:`functions` dictionary.

        Parameters
        ----------
        output_names: list
            The output names from the model
        Adds the loss functions to the :attr:`functions` dictionary.

        Parameters
        ----------
        output_names: list
            The output names from the model
        Adds the loss functions to the :attr:`functions` dictionary.

        Parameters
        ----------
        output_names: list
            The output names from the model
          """
          selected_loss = self._loss_dict[self._config["loss_function"]]
          mask_loss = self._loss_dict[self._config["mask_loss_function"]]
          loss_funcs = []
          for name in self._names:
keep keep keep keep replace replace replace replace keep keep keep keep keep
 training - Use explicit loss dictionaries to map to outputs 
plugins/train/model/_base.py