python argparse 模块

python argparse 模块

  • 利用argparse 传递参数
import argparse
def parse_args():
    parser = argparse.ArgumentParser(description = 'learn parse make firstly')  # 参数描述 
    parser.add_argument('--gpu',dest = 'gpu_id',             # --gpu 运行python文件时加在文件名后面参数 ,dest是别名,程序里调用
                       help = 'GPU device id to use [1]',           # help 提示信息                                      
                       default=0,type=int)                                  #  default m=默认值 type 是输入数据类型
    parser.add_argument('--solver', dest='solver',
                        help='solver prototxt',
                        default=None, type=str)
    args = parser.parse_args()
    return args
if __name__ == '__main__':
    args = parse_args()
    print(args)
    
;