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: ...@@ -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
......
...@@ -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
......
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