自己写一个类,模拟命令行参数

1,命令行参数

#parser = argparse.ArgumentParser()
#parser.add_argument('dataroot', help='path to dataset of kaggle ultrasound nerve segmentation')
## parser.add_argument('dataroot', default='data', help='path to dataset')
#parser.add_argument('--workers', type=int, help='number of data loading workers', default=1)
#parser.add_argument('--batchSize', type=int, default=64, help='input batch size')
#parser.add_argument('--niter', type=int, default=25, help='number of epochs to train for')
#parser.add_argument('--start_epoch', type=int, default=0, help='number of epoch to start')
#parser.add_argument('--lr', type=float, default=0.0002, help='learning rate, default=0.0002')
#parser.add_argument('--cuda'  , action='store_true', help='enables cuda')
#parser.add_argument('--resume', default='', type=str, metavar='PATH', help='path to latest checkpoint (default: none)')
#parser.add_argument('--useBN', action='store_true', help='enalbes batch normalization')
#parser.add_argument('--output_name', default='checkpoint___.tar', type=str, help='output checkpoint filename')

2,自己写一个类,模拟python中的命令行参数
class my_args:
    def __init__(self,
          dataroot = "./data/",
          workers = 4,
          batchSize = 4,
          niter = 5,
          start_epoch = 0,
          lr = 0.0002,
          cuda = True,
          resume ='checkpoint_BN.pkl',
          useBN = True,
          output_name = "checkpoint_BN.pkl"
          ):
        self.dataroot = dataroot
        self.workers = workers# windows bug
        self.batchSize = batchSize
        self.niter = niter
        self.start_epoch = start_epoch
        self.lr = lr
        self.cuda = cuda
        self.resume = resume
#        self.resume =''
        self.useBN = useBN
        self.output_name = output_name

3,生成类对象
#args = parser.parse_args()
args = my_args()