[Enhance] Add support for sequence input in --search-args option

mim/utils/utils.py

@@ -12,7 +12,7 @@
 from collections import defaultdict
 from email.parser import FeedParser
 from pkg_resources import get_distribution, parse_version
-from typing import Any, List, Optional, Tuple, Union
+from typing import Any, List, Optional, Sequence, Tuple, Union
 
 import click
 import requests
@@ -509,8 +509,11 @@ def get_config(cfg, name):
     name = name.split('.')
     suffix = ''
     for item in name:
-        assert item in cfg, f'attribute {item} not cfg{suffix}'
-        cfg = cfg[item]
+        if isinstance(cfg, Sequence) and not isinstance(cfg, str):
+            cfg = cfg[int(item)]
+        else:
+            assert item in cfg, f'attribute {item} not cfg{suffix}'
+            cfg = cfg[item]
         suffix += f'.{item}'
     return cfg
 
@@ -524,8 +527,11 @@ def set_config(cfg, name, value):
     name = name.split('.')
     suffix = ''
     for item in name[:-1]:
-        assert item in cfg, f'attribute {item} not cfg{suffix}'
-        cfg = cfg[item]
+        if isinstance(cfg, Sequence) and not isinstance(cfg, str):
+            cfg = cfg[int(item)]
+        else:
+            assert item in cfg, f'attribute {item} not cfg{suffix}'
+            cfg = cfg[item]
         suffix += f'.{item}'
 
     assert name[-1] in cfg, f'attribute {item} not cfg{suffix}'

tests/data/lenet5_mnist_2.0.py

@@ -12,7 +12,11 @@
 dataset_type = 'MNIST'
 data_preprocessor = dict(mean=[33.46], std=[78.87])
 
-pipeline = [dict(type='Resize', scale=32), dict(type='PackClsInputs')]
+pipeline = [
+    dict(type='Resize', scale=32),
+    dict(type='Pad', size=(32, 32)),
+    dict(type='PackClsInputs')
+]
 
 common_data_cfg = dict(
     type=dataset_type, data_prefix='data/mnist', pipeline=pipeline)

tests/test_gridsearch.py

@@ -57,6 +57,12 @@ def test_gridsearch(gpus, tmp_path):
         f'--work-dir={tmp_path}', '--search-args'
     ]
 
+    args5 = [
+        'mmcls', 'tests/data/lenet5_mnist_2.0.py', f'--gpus={gpus}',
+        f'--work-dir={tmp_path}', '--search-args',
+        '--train_dataloader.dataset.pipeline.0.scale 16 32'
+    ]
+
     result = runner.invoke(gridsearch, args1)
     assert result.exit_code == 0
 
@@ -69,6 +75,9 @@ def test_gridsearch(gpus, tmp_path):
     result = runner.invoke(gridsearch, args4)
     assert result.exit_code != 0
 
+    result = runner.invoke(gridsearch, args5)
+    assert result.exit_code == 0
+
 
 def teardown_module():
     runner = CliRunner()