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
Hide 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:
...
@@ -58,7 +58,7 @@ jobs:
-
script
:
|
-
script
:
|
python3 -m pip install torch==0.4.1 --user
python3 -m pip install torch==0.4.1 --user
python3 -m pip install torchvision==0.2.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'
displayName
:
'
Install
dependencies
for
integration'
-
script
:
|
-
script
:
|
source install.sh
source install.sh
...
...
src/sdk/pynni/nni/gp_tuner/gp_tuner.py
View file @
313b0f67
...
@@ -83,7 +83,7 @@ class GPTuner(Tuner):
...
@@ -83,7 +83,7 @@ class GPTuner(Tuner):
"""
"""
self
.
_space
=
TargetSpace
(
search_space
,
self
.
_random_state
)
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
"""Generate next parameter for trial
If the number of trial result is lower than cold start number,
If the number of trial result is lower than cold start number,
gp will first randomly generate some parameters.
gp will first randomly generate some parameters.
...
@@ -123,7 +123,7 @@ class GPTuner(Tuner):
...
@@ -123,7 +123,7 @@ class GPTuner(Tuner):
logger
.
info
(
"Generate paramageters:
\n
%s"
,
results
)
logger
.
info
(
"Generate paramageters:
\n
%s"
,
results
)
return
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.
"""Tuner receive result from trial.
Parameters
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