Unverified Commit 10d51b43 authored by xuehui's avatar xuehui Committed by GitHub
Browse files

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
parent a0820dd5
...@@ -58,7 +58,7 @@ export namespace ValidationSchemas { ...@@ -58,7 +58,7 @@ export namespace ValidationSchemas {
searchSpace: joi.string().required(), searchSpace: joi.string().required(),
maxExecDuration: joi.number().min(0).required(), maxExecDuration: joi.number().min(0).required(),
tuner: joi.object({ 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(), codeDir: joi.string(),
classFileName: joi.string(), classFileName: joi.string(),
className: joi.string(), className: joi.string(),
......
...@@ -24,6 +24,7 @@ ModuleName = { ...@@ -24,6 +24,7 @@ ModuleName = {
'Anneal': 'nni.hyperopt_tuner.hyperopt_tuner', 'Anneal': 'nni.hyperopt_tuner.hyperopt_tuner',
'Evolution': 'nni.evolution_tuner.evolution_tuner', 'Evolution': 'nni.evolution_tuner.evolution_tuner',
'SMAC': 'nni.smac_tuner.smac_tuner', 'SMAC': 'nni.smac_tuner.smac_tuner',
'BatchTuner': 'nni.batch_tuner.batch_tuner',
'Medianstop': 'nni.medianstop_assessor.medianstop_assessor' 'Medianstop': 'nni.medianstop_assessor.medianstop_assessor'
} }
...@@ -34,6 +35,7 @@ ClassName = { ...@@ -34,6 +35,7 @@ ClassName = {
'Anneal': 'HyperoptTuner', 'Anneal': 'HyperoptTuner',
'Evolution': 'EvolutionTuner', 'Evolution': 'EvolutionTuner',
'SMAC': 'SMACTuner', 'SMAC': 'SMACTuner',
'BatchTuner': 'BatchTuner',
'Medianstop': 'MedianstopAssessor' 'Medianstop': 'MedianstopAssessor'
} }
......
...@@ -31,7 +31,7 @@ Optional('maxTrialNum'): And(int, lambda x: 1 <= x <= 99999), ...@@ -31,7 +31,7 @@ Optional('maxTrialNum'): And(int, lambda x: 1 <= x <= 99999),
Optional('searchSpacePath'): os.path.exists, Optional('searchSpacePath'): os.path.exists,
'useAnnotation': bool, 'useAnnotation': bool,
'tuner': Or({ 'tuner': Or({
'builtinTunerName': Or('TPE', 'Random', 'Anneal', 'Evolution', 'SMAC'), 'builtinTunerName': Or('TPE', 'Random', 'Anneal', 'Evolution', 'SMAC', 'BatchTuner'),
'classArgs': { 'classArgs': {
'optimize_mode': Or('maximize', 'minimize'), 'optimize_mode': Or('maximize', 'minimize'),
Optional('speed'): int Optional('speed'): int
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment