                  out.write(state_json.encode("utf-8"))
          except IOError as err:
              logger.error("Unable to save model state: %s", str(err.strerror))
          logger.debug("Saved State")
  
      def backup(self):
          """ Backup state file """
          origfile = self.filename
 
</s> remove     def save(self, should_backup=False):
 </s> add     def save(self, should_backup=False, snapshot=False): </s> remove     def save_models(self):
 </s> add     def save_models(self, snapshot_iteration): </s> remove                 model.save_models()
 </s> add                 model.save_models(False) </s> remove         print("\n")
 </s> add 
        if snapshot:
            self.snapshot_model()
                  out.write(state_json.encode("utf-8"))
          except IOError as err:
              logger.error("Unable to save model state: %s", str(err.strerror))
          logger.debug("Saved State")
        if snapshot:
            self.snapshot_model()
  
      def backup(self):
          """ Backup state file """
          origfile = self.filename
keep keep keep add keep keep keep keep
 Add save snapshot backup option for training 
plugins/train/model/_base.py