10.3.8. 通过命令行覆盖config参数

HAT支持通过命令行修改config参数，使用方式，在 --opts 后面加上 (key, value) 键值对即可，如下：

python tools/train.py --stage float --config configs/classification/resnet18.py --opts "key1" "value1" "key2" "value2"

其中， key 是要修改的参数名， value 为想要传入的值。

例如配置文件 config.py ：

task_name = "resnet_cls"
num_classes=1000
devices = [0, 1, 2, 3]
data_shape = (3, 224, 224)
model = dict(
    backbone="resnet18",
    num_classes=num_classes,
)

注解

本文档中的示例，仅为向您说明使用 --opts 时的不同类型参数的正确书写格式，示例 config 中的参数、字段配置等，可能与项目中实际 config 略有不同。

• key 可以支持多级参数的修改，如下所示:

  python tools/train.py --stage float --config config.py --opts model.backbone "'resnet50'"
  

  上面这条命令，可以把 config 中 model.backbone 字段的值修改成"resnet50" 。

• value 可以是数值、字符串（str）、列表（list）、元组（tuple）形式，但不支持 dict 形式。

  如下命令，执行后可以将 config 中 model.num_classes 字段的值修改成10。

  python tools/train.py --stage float --config config.py --opts num_classes 10
  

  但是由于解析机制的原因，在要传入 value 是 tuple 或 list 形式时，需要对 tuple 或 list 加上引号：

  python tools/train.py --stage float --config config.py --opts data_shape (3, 112, 112)  # 这样会解析失败
  python tools/train.py --stage float --config config.py --opts data_shape "(3, 112, 112)" # 这样是正确的
  

  如果要传入的 value 是 str 形式的，需要额外加上引号，例如：

  python tools/train.py --stage float --config config.py --opts model.backbone resnet18 # 这样会解析不正确
  python tools/train.py --stage float --config config.py --opts model.backbone "resnet18" # 这样会解析不正确
  python tools/train.py --stage float --config config.py --opts model.backbone "'resnet18'" # 这样是正确的