有时候想在命令行中输入参数如 python try.py filenameA
教程:https://blog.csdn.net/yskyskyer123/article/details/80667861
如下是该教程中的部分代码
#option.py
import argparse
class Option():
def __init__(self):
self.parser = argparse.ArgumentParser(description='PyTorch Light CNN Training')
self.parser.add_argument('--no_cuda','-nc',action='store_false')
self.parser.add_argument('--root_path',type=str,default='photo2sketch/trainA')
self.parser.add_argument('--workers',type=int,default=16)
self.parser.add_argument('--batch_size',type=int,default=1)
self.parser.add_argument('--start_epoch',type=int,default=1)
self.parser.add_argument('--end_epoch',type=int,default=100)
self.parser.add_argument('--train_list',type=str,default='info.txt')
def parse(self):
opt=self.parser.parse_args()
return opt
#使用并解析参数
opt=Option().parse()
print(opt.batch_size)