OpenExplorer
地平线官网
地平线智能汽车开发者社区
地平线官网
地平线智能汽车开发者社区
中文

通过命令行覆盖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'" # 这样是正确的