def get_args()

in PairSupCon/main.py [0:0]

• 32 lines of code
• 1 McCabe index (conditional complexity)

def get_args(argv):
    parser = argparse.ArgumentParser()
    parser.add_argument('--training_instance', type=str, default='local')
    parser.add_argument('--gpuid', nargs="+", type=int, default=[0], help="The list of gpuid, ex:--gpuid 3 1. Negative value means cpu-only")
    parser.add_argument('--seed', type=int, default=0, help="")
    parser.add_argument('--resdir', type=str, default='./results')
    parser.add_argument('--logging_step', type=int, default=250, help="")
    parser.add_argument('--dev_set', default="None", help="use sts-b as dev set or not", choices=["None", "sts"])  
    parser.add_argument('--path_sts_data', type=str, default='', help="use sts-b as dev set")
    parser.add_argument('--s3_ckptdir', type=str, default='', help="s3path for ckpts")  
    # Dataset
    parser.add_argument('--datapath', type=str, default='../datasets/NLI/')
    parser.add_argument('--dataname', type=str, default='nli_pairsupcon.csv', help="")
    parser.add_argument('--num_classes', type=int, default=2)
    parser.add_argument('--text', type=str, default='text')
    parser.add_argument('--pairsimi', type=str, default='pairsimi')
    # Training parameters
    parser.add_argument('--max_length', type=int, default=32)
    parser.add_argument('--batch_size', type=int, default=1024)
    parser.add_argument('--lr', type=float, default=5e-06, help="")
    parser.add_argument('--lr_scale', type=int, default=100, help="")
    parser.add_argument('--epochs', type=int, default=3)
    parser.add_argument('--max_iter', type=int, default=100000000)
    # Contrastive learning
    parser.add_argument('--mode', type=str, default='pairsupcon', help="")
    parser.add_argument('--bert', type=str, default='bertbase', choices=["bertbase", "bertlarge"], help="")
    parser.add_argument('--contrast_type', type=str, default="HardNeg")
    parser.add_argument('--feat_dim', type=int, default=128, help="dimension of the projected features for instance discrimination loss")
    parser.add_argument('--temperature', type=float, default=0.05, help="temperature required by contrastive loss")
    parser.add_argument('--beta', type=float, default=1, help=" ")
    
    args = parser.parse_args(argv)
    args.use_gpu = args.gpuid[0] >= 0
    args.resPath = None
    args.tensorboard = None
    return args