code: def save(self, model_file, save_dir, verbose=True):
          
  
          # Check existence of model saving directory and create if does not exist.
          if not os.path.exists(save_dir):
              os.makedirs(save_dir)
  
          params = {
              "model": self.state_dict(),
              "cardinality": self.cardinality,
              "name": self.name,
              "config": self.settings,
          }
  
          try:
              torch.save(params, f"{save_dir}/{model_file}")
          except BaseException:
              self.logger.warning("Saving failed... continuing anyway.")
  
          if verbose:
              self.logger.info(f"[{self.name}] Model saved as {model_file} in {save_dir}")
docs: Save current model.
  
          :param model_file: Saved model file name.
          :type model_file: str
          :param save_dir: Saved model directory.
          :type save_dir: str
          :param verbose: Print log or not
          :type verbose: bool