parser.add_argument:action的作用

燕意蕴
2023-12-01
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

 类似资料: