Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
31b2b886
Commit
31b2b886
authored
Nov 25, 2019
by
liuzhe-lz
Committed by
chicm-ms
Nov 25, 2019
Browse files
Fix bug introduced in customized trial (#1779)
parent
48084d88
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
6 additions
and
1 deletion
+6
-1
src/sdk/pynni/nni/msg_dispatcher.py
src/sdk/pynni/nni/msg_dispatcher.py
+1
-1
test/config_test/multi_thread/multi_thread_tuner.py
test/config_test/multi_thread/multi_thread_tuner.py
+5
-0
No files found.
src/sdk/pynni/nni/msg_dispatcher.py
View file @
31b2b886
...
@@ -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
:
...
...
test/config_test/multi_thread/multi_thread_tuner.py
View file @
31b2b886
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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment