Update batch tuner (#158)

* update readme in ga_squad * update readme * fix typo * Update README.md * Update README.md * Update README.md * update readme * update batch tuner
microsoft · Oct 8, 2018 · 10d51b4 · 10d51b4
1 parent a0820dd
commit 10d51b4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/src/nni_manager/rest_server/restValidationSchemas.ts b/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(),

diff --git a/src/sdk/pynni/nni/constants.py b/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'
 }

diff --git a/tools/nnicmd/config_schema.py b/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