              logger.debug("IOError: %s", str(err))
          except JSONDecodeError as err:
              logger.debug("JSONDecodeError: %s:", str(err))
  
      def save(self, should_backup=False):
          """ Save iteration number to state file """
          logger.debug("Saving State")
          if should_backup:
              self.backup()
          try:
 
</s> remove                 model.save_models()
 </s> add                 model.save_models(False) </s> remove         print("\n")
 </s> add 
    def save(self, should_backup=False, snapshot=False):
              logger.debug("IOError: %s", str(err))
          except JSONDecodeError as err:
              logger.debug("JSONDecodeError: %s:", str(err))
  
    def save(self, should_backup=False, snapshot=False):
          """ Save iteration number to state file """
          logger.debug("Saving State")
          if should_backup:
              self.backup()
          try:
keep keep keep keep replace keep keep keep keep keep
 Add save snapshot backup option for training 
plugins/train/model/_base.py