Commit 31b2b886 authored by liuzhe-lz's avatar liuzhe-lz Committed by chicm-ms
Browse files

Fix bug introduced in customized trial (#1779)

parent 48084d88
...@@ -184,7 +184,7 @@ class MsgDispatcher(MsgDispatcherBase): ...@@ -184,7 +184,7 @@ class MsgDispatcher(MsgDispatcherBase):
""" """
id_ = data['parameter_id'] id_ = data['parameter_id']
value = data['value'] value = data['value']
if not id_ or id_ in _customized_parameter_ids: if id_ is None or id_ in _customized_parameter_ids:
if not hasattr(self.tuner, '_accept_customized'): if not hasattr(self.tuner, '_accept_customized'):
self.tuner._accept_customized = False self.tuner._accept_customized = False
if not self.tuner._accept_customized: if not self.tuner._accept_customized:
......
import logging
import time import time
from nni.tuner import Tuner from nni.tuner import Tuner
...@@ -7,14 +8,18 @@ class MultiThreadTuner(Tuner): ...@@ -7,14 +8,18 @@ class MultiThreadTuner(Tuner):
self.parent_done = False self.parent_done = False
def generate_parameters(self, parameter_id, **kwargs): def generate_parameters(self, parameter_id, **kwargs):
logging.debug('generate_parameters: %s %s', parameter_id, kwargs)
if parameter_id == 0: if parameter_id == 0:
return {'x': 0} return {'x': 0}
else: else:
while not self.parent_done: while not self.parent_done:
logging.debug('parameter_id %s sleeping', parameter_id)
time.sleep(2) time.sleep(2)
logging.debug('parameter_id %s waked up', parameter_id)
return {'x': 1} return {'x': 1}
def receive_trial_result(self, parameter_id, parameters, value, **kwargs): def receive_trial_result(self, parameter_id, parameters, value, **kwargs):
logging.debug('receive_trial_result: %s %s %s %s', parameter_id, parameters, value, kwargs)
if parameter_id == 0: if parameter_id == 0:
self.parent_done = True self.parent_done = True
......
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