def METHOD_NAME(self, output_dir: str):
    """
    The function saves the model to the path specified, and also saves the label2id and id2label
    dictionaries to the same path
    Args:
      output_dir (str): the path to save the model to
    """
    self._span_marker_model.save_pretrained(output_dir)