update

src/sdk/pynni/nni/batch_tuner/batch_tuner.py

@@ -47,7 +47,7 @@ class BatchTuner(Tuner):
     }
     '''
 
-    def __init__(self, optimize_mode):
+    def __init__(self):
         self.count = -1
         self.values = []
 

src/sdk/pynni/nni/gridsearch_tuner/gridsearch_tuner.py

@@ -48,7 +48,7 @@ class GridSearchTuner(Tuner):
     and sample and then change the sampled value back.
     '''
 
-    def __init__(self, optimize_mode):
+    def __init__(self):
         self.count = -1
         self.expanded_search_space = []
 

test/sdk_test.py

@@ -29,7 +29,7 @@
 RED = '\33[31m'
 CLEAR = '\33[0m'
 
-TUNER_LIST = ['BatchTuner', 'TPE', 'Random', 'Anneal', 'Evolution']
+TUNER_LIST = ['GridSearch', 'BatchTuner', 'TPE', 'Random', 'Anneal', 'Evolution']
 ASSESSOR_LIST = ['Medianstop']
 EXPERIMENT_URL = 'http://localhost:8080/api/v1/nni/experiment'
 
@@ -38,12 +38,17 @@ def switch(dispatch_type, dispatch_name):
     '''Change dispatch in config.yml'''
     config_path = 'sdk_test/local.yml'
     experiment_config = get_yml_content(config_path)
-    experiment_config[dispatch_type.lower()] = {
-        'builtin' + dispatch_type + 'Name': dispatch_name,
-        'classArgs': {
-            'optimize_mode': 'maximize'
+    if dispatch_name in ['GridSearch', 'BatchTuner']:
+        experiment_config[dispatch_type.lower()] = {
+            'builtin' + dispatch_type + 'Name': dispatch_name
+        }
+    else:
+        experiment_config[dispatch_type.lower()] = {
+            'builtin' + dispatch_type + 'Name': dispatch_name,
+            'classArgs': {
+                'optimize_mode': 'maximize'
+            }
         }
-    }
     dump_yml_content(config_path, experiment_config)
 
 def test_builtin_dispatcher(dispatch_type, dispatch_name):

tools/nni_cmd/config_schema.py

@@ -34,8 +34,13 @@
 Optional('multiThread'): bool,
 'useAnnotation': bool,
 'tuner': Or({
-    'builtinTunerName': Or('TPE', 'Random', 'Anneal', 'Evolution', 'SMAC', 'BatchTuner', 'GridSearch'),
-    Optional('classArgs'): dict,
+    'builtinTunerName': Or('TPE', 'Random', 'Anneal', 'SMAC', 'Evolution'),
+    Optional('classArgs'): {
+        'optimize_mode': Or('maximize', 'minimize')
+    },
+    Optional('gpuNum'): And(int, lambda x: 0 <= x <= 99999),
+},{
+    'builtinTunerName': Or('BatchTuner', 'GridSearch'),
     Optional('gpuNum'): And(int, lambda x: 0 <= x <= 99999),
 },{
     'codeDir': os.path.exists,
@@ -46,7 +51,10 @@
 }),
 Optional('assessor'): Or({
     'builtinAssessorName': lambda x: x in ['Medianstop'],
-    Optional('classArgs'): dict,
+    Optional('classArgs'): {
+        Optional('optimize_mode'): Or('maximize', 'minimize'),
+        Optional('start_step'): And(int, lambda x: 0 <= x <= 9999)
+    },
     Optional('gpuNum'): And(int, lambda x: 0 <= x <= 99999)
 },{
     'codeDir': os.path.exists,