INNER CODE UNIT · Python
_setup_parser
zjunlp/PromptKG · research/GenKGC/main.py:30
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="KGC")
parser.add_argument("--chunk", type=str, default="")
parser.add_argument("--model_class", type=str, default="RobertaUseLabelWord")
parser.add_argument("--checkpoint", type=str, default=None)