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
bc44c745
"vscode:/vscode.git/clone" did not exist on "2a48812edb1a7c3e280159637fa89b7a0bbfb86b"
Commit
bc44c745
authored
Nov 30, 2018
by
Zejun Lin
Committed by
QuanluZhang
Nov 30, 2018
Browse files
fix sdk for NoMoreTrial status (#394)
* fix bug * add docs
parent
d33220f7
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
4 deletions
+5
-4
src/sdk/pynni/nni/msg_dispatcher.py
src/sdk/pynni/nni/msg_dispatcher.py
+3
-4
src/sdk/pynni/nni/tuner.py
src/sdk/pynni/nni/tuner.py
+2
-0
No files found.
src/sdk/pynni/nni/msg_dispatcher.py
View file @
bc44c745
...
@@ -99,12 +99,11 @@ class MsgDispatcher(MsgDispatcherBase):
...
@@ -99,12 +99,11 @@ class MsgDispatcher(MsgDispatcherBase):
ids
=
[
_create_parameter_id
()
for
_
in
range
(
data
)]
ids
=
[
_create_parameter_id
()
for
_
in
range
(
data
)]
params_list
=
self
.
tuner
.
generate_multiple_parameters
(
ids
)
params_list
=
self
.
tuner
.
generate_multiple_parameters
(
ids
)
for
i
,
_
in
enumerate
(
params_list
):
send
(
CommandType
.
NewTrialJob
,
_pack_parameter
(
ids
[
i
],
params_list
[
i
]))
# when parameters is None.
# when parameters is None.
if
len
(
params_list
)
==
0
:
if
len
(
params_list
)
<
len
(
ids
)
:
send
(
CommandType
.
NoMoreTrialJobs
,
_pack_parameter
(
ids
[
0
],
''
))
send
(
CommandType
.
NoMoreTrialJobs
,
_pack_parameter
(
ids
[
0
],
''
))
else
:
for
i
,
_
in
enumerate
(
ids
):
send
(
CommandType
.
NewTrialJob
,
_pack_parameter
(
ids
[
i
],
params_list
[
i
]))
return
True
return
True
def
handle_update_search_space
(
self
,
data
):
def
handle_update_search_space
(
self
,
data
):
...
...
src/sdk/pynni/nni/tuner.py
View file @
bc44c745
...
@@ -41,6 +41,8 @@ class Tuner(Recoverable):
...
@@ -41,6 +41,8 @@ class Tuner(Recoverable):
"""Returns multiple sets of trial (hyper-)parameters, as iterable of serializable objects.
"""Returns multiple sets of trial (hyper-)parameters, as iterable of serializable objects.
Call 'generate_parameters()' by 'count' times by default.
Call 'generate_parameters()' by 'count' times by default.
User code must override either this function or 'generate_parameters()'.
User code must override either this function or 'generate_parameters()'.
If there's no more trial, user should raise nni.NoMoreTrialError exception in generate_parameters().
If so, this function will only return sets of trial (hyper-)parameters that have already been collected.
parameter_id_list: list of int
parameter_id_list: list of int
"""
"""
result
=
[]
result
=
[]
...
...
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