关于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

发布了270 篇原创文章 · 获赞 11 · 访问量 2万+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章