深度学习参数设置（基于argparse）

定义参数

#更多使用方法
#https://docs.python.org/zh-cn/3/library/argparse.html#argumentparser-objects
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--config', type=str, default='configs/edges2shoes_folder.yaml', help='Path to the config file.')
parser.add_argument('--resume', action="store_true",help='resume train')        #不触发为假，触发为真
parser.add_argument('--action',choices=['train','test'])                        #只能输入choices中的值否则报错
args = parser.parse_args()

print(args.config)
if args.resume :
    print('yes')
else:
    print('no')


#执行
#(pythonProject) D:\Desktop\MUNIT-master>python Mytest.py --action test
#configs/edges2shoes_folder.yaml
#no