Unverified Commit 313b0f67 authored by chicm-ms's avatar chicm-ms Committed by GitHub
Browse files

Fix gp tuner (#1592)

* fix gp tuner
parent e93d2c25
......@@ -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
......
......@@ -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
......
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