Commit 54118c34 authored by demianzhang's avatar demianzhang Committed by xuehui
Browse files

Fix BatchTuner test (#981)

* Update naive_trial.py

* Fix test BatchTuner
parent dd04c73a
...@@ -49,6 +49,10 @@ def switch(dispatch_type, dispatch_name): ...@@ -49,6 +49,10 @@ def switch(dispatch_type, dispatch_name):
'optimize_mode': 'maximize' 'optimize_mode': 'maximize'
} }
} }
if dispatch_name == 'BatchTuner':
experiment_config['searchSpacePath'] = 'batchtuner_search_space.json'
else:
experiment_config['searchSpacePath'] = 'search_space.json'
dump_yml_content(config_path, experiment_config) dump_yml_content(config_path, experiment_config)
def test_builtin_dispatcher(dispatch_type, dispatch_name): def test_builtin_dispatcher(dispatch_type, dispatch_name):
......
{
"combine_params":
{
"_type" : "choice",
"_value" : [{"x": 1},
{"x": 100}]
}
}
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