parser.add_argument:action的作用

在PyTorch中,`--no-cuda`参数是一个方便的开关,用于决定是否在训练过程中禁用CUDA。当指定`--no-cuda`时,即使系统有GPU,训练也会在CPU上进行。默认情况下,该参数为False,意味着如果系统支持,将使用CUDA。通过在命令行添加`--no-cuda`,可以强制程序不使用GPU,这对于没有GPU的环境或想在CPU上运行代码的情况非常有用。
摘要由CSDN通过智能技术生成
 
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

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值