                  self.training_size = state.get("training_size", 256)
                  self.inputs = state.get("inputs", dict())
                  self.config = state.get("config", dict())
                  logger.debug("Loaded state: %s", state)
                  self.update_legacy_config()
                  self.replace_config(config_changeable_items)
          except IOError as err:
              logger.warning("No existing state file found. Generating.")
              logger.debug("IOError: %s", str(err))
          except JSONDecodeError as err:
 
</s> remove         self.funcs = self.get_loss_functions(side, predict, mask_input)
 </s> add         self.funcs = self.get_loss_functions(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):
                  self.training_size = state.get("training_size", 256)
                  self.inputs = state.get("inputs", dict())
                  self.config = state.get("config", dict())
                  logger.debug("Loaded state: %s", state)

                  self.replace_config(config_changeable_items)
          except IOError as err:
              logger.warning("No existing state file found. Generating.")
              logger.debug("IOError: %s", str(err))
          except JSONDecodeError as err:
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