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
313b0f67
Unverified
Commit
313b0f67
authored
Oct 09, 2019
by
chicm-ms
Committed by
GitHub
Oct 09, 2019
Browse files
Fix gp tuner (#1592)
* fix gp tuner
parent
e93d2c25
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
3 deletions
+3
-3
azure-pipelines.yml
azure-pipelines.yml
+1
-1
src/sdk/pynni/nni/gp_tuner/gp_tuner.py
src/sdk/pynni/nni/gp_tuner/gp_tuner.py
+2
-2
No files found.
azure-pipelines.yml
View file @
313b0f67
...
...
@@ -58,7 +58,7 @@ jobs:
-
script
:
|
python3 -m pip install torch==0.4.1 --user
python3 -m pip install torchvision==0.2.1 --user
python3 -m pip install tensorflow --user
python3 -m pip install tensorflow
==1.13.1
--user
displayName
:
'
Install
dependencies
for
integration'
-
script
:
|
source install.sh
...
...
src/sdk/pynni/nni/gp_tuner/gp_tuner.py
View file @
313b0f67
...
...
@@ -83,7 +83,7 @@ class GPTuner(Tuner):
"""
self
.
_space
=
TargetSpace
(
search_space
,
self
.
_random_state
)
def
generate_parameters
(
self
,
parameter_id
):
def
generate_parameters
(
self
,
parameter_id
,
**
kwargs
):
"""Generate next parameter for trial
If the number of trial result is lower than cold start number,
gp will first randomly generate some parameters.
...
...
@@ -123,7 +123,7 @@ class GPTuner(Tuner):
logger
.
info
(
"Generate paramageters:
\n
%s"
,
results
)
return
results
def
receive_trial_result
(
self
,
parameter_id
,
parameters
,
value
):
def
receive_trial_result
(
self
,
parameter_id
,
parameters
,
value
,
**
kwargs
):
"""Tuner receive result from trial.
Parameters
...
...
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