          logger.debug("Initializing %s: (side: '%s', outputs: '%s', mask_input: '%s', predict: %s",
                       self.__class__.__name__, side, outputs, mask_input, predict)
          self.outputs = outputs
          self.names = self.get_loss_names()
          self.funcs = self.get_loss_functions(side, predict, mask_input)
          if len(self.names) > 1:
              self.names.insert(0, "total_loss")
          logger.debug("Initialized: %s", self.__class__.__name__)
  
      @property
 
</s> remove     def __init__(self, side, outputs, mask_input, predict):
        logger.debug("Initializing %s: (side: '%s', outputs: '%s', mask_input: '%s', predict: %s",
                     self.__class__.__name__, side, outputs, mask_input, predict)
 </s> add     def __init__(self, outputs, mask_input):
        logger.debug("Initializing %s: (outputs: '%s', mask_input: '%s')",
                     self.__class__.__name__, outputs, mask_input) </s> remove             loss = Loss(side, model.outputs, mask_input, self.predict)
 </s> add             loss = Loss(model.outputs, mask_input) </s> remove     def get_loss_functions(self, side, predict, mask):
 </s> add     def get_loss_functions(self, mask): </s> remove                 self.update_legacy_config()
 </s> add 
        self.funcs = self.get_loss_functions(mask_input)
          logger.debug("Initializing %s: (side: '%s', outputs: '%s', mask_input: '%s', predict: %s",
                       self.__class__.__name__, side, outputs, mask_input, predict)
          self.outputs = outputs
          self.names = self.get_loss_names()
        self.funcs = self.get_loss_functions(mask_input)
          if len(self.names) > 1:
              self.names.insert(0, "total_loss")
          logger.debug("Initialized: %s", self.__class__.__name__)
  
      @property
keep keep keep keep replace keep keep keep keep keep
 Post PR Fixups

- train/_config.py: PEP8 Fixes. Slight description change on coverage
- models/_base.py:
   - Remove unused variables from Loss()
   - Delete legacy config items from state file
   - Save state file on Legacy update
   - PEP8
- Remove _defaults.py for models with no config options 
plugins/train/model/_base.py