pytorch-argparse的用法

import argparse
def parse():
    parser = argparse.ArgumentParser()
    parser.add_argument('--scales',help='scales',type=int,default=5)
    opt = parser.parse_args()
    return opt
if __name__ == '__main__':
    opt = parse()
    print(opt)
    opt.num = 'now two args'
    print(opt)

  结果:

Namespace(scales=5)

Namespace(num='now two args', scales=5)

note:使用argparse加载参数,在parse()方法中定义完opt后,其他地方再用到其他参数,可以直接加参数,例如

opt.num = 'now two args'