關於python argparse的一些總結

import argparse



parser = argparse.ArgumentParser()
parser.add_argument("--task", default="pre-trained", help="path to folder containing the model")
parser.add_argument("--data_syn_dir", default="", help="path to synthetic dataset")
parser.add_argument("--data_real_dir", default="", help="path to real dataset")
parser.add_argument("--save_model_freq", default=1, type=int, help="frequency to save model")
parser.add_argument("--dd", default=1.0, type=float, help="frel")
parser.add_argument("--dd2", default=True, type=bool, help="frel")
parser.add_argument("--is_hyper", default=1, type=int, help="use hypercolumn or not")
parser.add_argument("--is_training", default=1, help="training or testing")
parser.add_argument("--continue_training", action="store_true", help="search for checkpoint in the subfolder specified by `task` argument")
ARGS = parser.parse_args()

if ARGS.continue_training:
    print("YES")
else:
    print("NO")

print(type(ARGS.dd),type(ARGS.task),type(ARGS.is_hyper),type(ARGS.continue_training),type(ARGS.dd2))

有點特殊的是 action='store_true'，這個是在命令行中 打出 python xxx.py --continue_training 則 continue_training=True,沒有打出則False