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
6f3a90eb
"src/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "5c0a78075e7ff97149a8567b90f32e97576c8b1e"
Unverified
Commit
6f3a90eb
authored
Feb 14, 2019
by
Yan Ni
Committed by
GitHub
Feb 14, 2019
Browse files
fix metis bug (#728)
parent
c7af5b73
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
6 deletions
+8
-6
src/sdk/pynni/nni/metis_tuner/metis_tuner.py
src/sdk/pynni/nni/metis_tuner/metis_tuner.py
+8
-6
No files found.
src/sdk/pynni/nni/metis_tuner/metis_tuner.py
View file @
6f3a90eb
...
@@ -25,6 +25,8 @@ import random
...
@@ -25,6 +25,8 @@ import random
import
statistics
import
statistics
import
sys
import
sys
import
numpy
as
np
from
enum
import
Enum
,
unique
from
enum
import
Enum
,
unique
from
multiprocessing.dummy
import
Pool
as
ThreadPool
from
multiprocessing.dummy
import
Pool
as
ThreadPool
...
@@ -120,7 +122,7 @@ class MetisTuner(Tuner):
...
@@ -120,7 +122,7 @@ class MetisTuner(Tuner):
self
.
x_types
[
idx
]
=
'range_int'
self
.
x_types
[
idx
]
=
'range_int'
else
:
else
:
bounds
=
[]
bounds
=
[]
for
value
in
range
(
key_range
[
0
],
key_range
[
1
],
key_range
[
2
]):
for
value
in
np
.
a
range
(
key_range
[
0
],
key_range
[
1
],
key_range
[
2
]):
bounds
.
append
(
value
)
bounds
.
append
(
value
)
self
.
x_bounds
[
idx
]
=
bounds
self
.
x_bounds
[
idx
]
=
bounds
self
.
x_types
[
idx
]
=
'discrete_int'
self
.
x_types
[
idx
]
=
'discrete_int'
...
@@ -177,7 +179,7 @@ class MetisTuner(Tuner):
...
@@ -177,7 +179,7 @@ class MetisTuner(Tuner):
minimize_starting_points
=
self
.
minimize_starting_points
,
minimize_starting_points
=
self
.
minimize_starting_points
,
minimize_constraints_fun
=
self
.
minimize_constraints_fun
)
minimize_constraints_fun
=
self
.
minimize_constraints_fun
)
logger
.
info
(
"Generate paramageters:
\n
"
,
str
(
results
))
logger
.
info
(
"Generate paramageters:
\n
%s
"
,
str
(
results
))
return
results
return
results
...
@@ -192,8 +194,8 @@ class MetisTuner(Tuner):
...
@@ -192,8 +194,8 @@ class MetisTuner(Tuner):
value
=
-
value
value
=
-
value
logger
.
info
(
"Received trial result."
)
logger
.
info
(
"Received trial result."
)
logger
.
info
(
"value is :
"
,
str
(
value
)
)
logger
.
info
(
"value is :
\t
%f"
,
value
)
logger
.
info
(
"parameter is :
"
,
str
(
parameters
))
logger
.
info
(
"parameter is :
\t
%s
"
,
str
(
parameters
))
# parse parameter to sample_x
# parse parameter to sample_x
sample_x
=
[
0
for
i
in
range
(
len
(
self
.
key_order
))]
sample_x
=
[
0
for
i
in
range
(
len
(
self
.
key_order
))]
...
...
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