跟着论文代码学习编码第一天：main.py

根据ESRT和LBNet的代码学习编码。首先看main.py。

1.  args模块  B站小侯学府的args讲解

需要三步，创建argparse.ArgumentParser解释器，添加add_argument参数，解析参数parse_args:

# 创建argparse.ArgumentParser解析器
parser = argparse.ArgumentParser(description='LBNet')

# 添加参数add_argument('添加的变量', type=变量类型, default=默认值, help='使用-h时告知的作用')
parser.add_argument('--n_threads', type=int, default=12,
                    help='number of threads for data loading')
parser.add_argument('--cpu', action='store_true',
                    help='use cpu only')


# 解析参数parse_args
args = parser.parse_args()      # 使用时 args.参数名 即可调用