Commit bfb4074d authored by xuehui's avatar xuehui
Browse files

update __main__.py

parent 693cf20f
...@@ -35,7 +35,7 @@ from nni.medianstop_assessor.medianstop_assessor import MedianstopAssessor ...@@ -35,7 +35,7 @@ from nni.medianstop_assessor.medianstop_assessor import MedianstopAssessor
logger = logging.getLogger('nni.main') logger = logging.getLogger('nni.main')
logger.debug('START') logger.debug('START')
BUILT_IN_CLASS_NAMES = ['HyperoptTuner', 'EvolutionTuner', 'MedianstopAssessor'] BUILT_IN_CLASS_NAMES = ['HyperoptTuner', 'EvolutionTuner', 'BatchTuner', 'MedianstopAssessor']
def create_builtin_class_instance(classname, jsonstr_args): def create_builtin_class_instance(classname, jsonstr_args):
if jsonstr_args: if jsonstr_args:
......
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