• 中文
  • 3.5.0

    • 通过命令行覆盖config参数

    HAT支持通过命令行修改config参数，在 --opts 后面加上 (key, value) 键值对即可。其中， key 是要修改的参数名， value 为想要传入的值。

    使用方法

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

    使用示例

    例如，我们有一个配置文件 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'" # 这样是正确的