parser.add_argument('--no-cuda', action='store_true', default=False,
help='Disable cuda for training')
有了 action
这一参数,就相当于把 --no-cuda
参数设成了一个“开关”。我们是不需要给这个开关传递具体的值的,只需要“拨一下”这个开关就行了。
比如上面action='store_true'
,就是说当拨动了这个“开关”,就将参数--no-cuda
存储为True
,否则参数会存储为False
,因为我们设置的 default=False
。
实例:
# 启动action
python demo.py --no-cuda
>>> --no-cuda=True
这样就相当于我们把 --no-cuda
这个开关拨到了True
上
# 不启动action
python demo.py
>>> --no-cuda=False
此时没有理 --no-cuda
这个开关,那么程序就会自动给它赋值为 False
。