"src/git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "5c0a78075e7ff97149a8567b90f32e97576c8b1e"
Unverified Commit 6f3a90eb authored by Yan Ni's avatar Yan Ni Committed by GitHub
Browse files

fix metis bug (#728)

parent c7af5b73
...@@ -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.arange(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))]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment