INNER CODE UNIT · Python
_setup_parser
zjunlp/PromptKG · research/RetrievalRE/main.py:24
def _setup_parser():
"""Set up Python's ArgumentParser with data, model, trainer, and other arguments."""
parser = argparse.ArgumentParser(add_help=False)
# Add Trainer specific arguments, such as --max_epochs, --gpus, --precision
trainer_parser = pl.Trainer.add_argparse_args(parser)
trainer_parser._action_groups[1].title = "Trainer Args" # pylint: disable=protected-access
parser = argparse.ArgumentParser(add_help=False, parents=[trainer_parser])
# Basic arguments
parser.add_argument("--wandb", action="store_true", default=False)
parser.add_argument("--litmodel_class", type=str, default="TransformerLitModel")
parser.add_argument("--seed", type=int, default=7)
parser.add_argument("--data_class", type=str, default="DIALOGUE")
parser.add_argument("--lr_2", type=float, default=3e-5)
parser.add_argument("--model_class", type=str, default="bert.BertForSequenceClassification")
parser.add_argument("--two_steps", default=False, action="store_true")
parser.add_argument("--load_checkpoint", type=str, default=None)