Update batch tuner

src/nni_manager/rest_server/restValidationSchemas.ts

@@ -58,7 +58,7 @@ export namespace ValidationSchemas {
             searchSpace: joi.string().required(),
             maxExecDuration: joi.number().min(0).required(),
             tuner: joi.object({
-                builtinTunerName: joi.string().valid('TPE', 'Random', 'Anneal', 'Evolution', 'SMAC'),
+                builtinTunerName: joi.string().valid('TPE', 'Random', 'Anneal', 'Evolution', 'SMAC', 'BatchTuner'),
                 codeDir: joi.string(),
                 classFileName: joi.string(),
                 className: joi.string(),

src/sdk/pynni/nni/constants.py

@@ -24,6 +24,7 @@
     'Anneal': 'nni.hyperopt_tuner.hyperopt_tuner',
     'Evolution': 'nni.evolution_tuner.evolution_tuner',
     'SMAC': 'nni.smac_tuner.smac_tuner',
+    'BatchTuner': 'nni.batch_tuner.batch_tuner',
 
     'Medianstop': 'nni.medianstop_assessor.medianstop_assessor'
 }
@@ -34,6 +35,7 @@
     'Anneal': 'HyperoptTuner',
     'Evolution': 'EvolutionTuner',
     'SMAC': 'SMACTuner',
+    'BatchTuner': 'BatchTuner',
 
     'Medianstop': 'MedianstopAssessor'
 }

tools/nnicmd/config_schema.py

@@ -31,7 +31,7 @@
 Optional('searchSpacePath'): os.path.exists,
 'useAnnotation': bool,
 'tuner': Or({
-    'builtinTunerName': Or('TPE', 'Random', 'Anneal', 'Evolution', 'SMAC'),
+    'builtinTunerName': Or('TPE', 'Random', 'Anneal', 'Evolution', 'SMAC', 'BatchTuner'),
     'classArgs': {
         'optimize_mode': Or('maximize', 'minimize'),
         Optional('speed'): int