Commit df3952a2 authored by suiguoxin's avatar suiguoxin
Browse files

fix val err

parent dbac6edb
authorName: default
experimentName: example_auto-gbdt
trialConcurrency: 1
maxExecDuration: 10h
maxTrialNum: 10
maxExecDuration: 48h
maxTrialNum: 1000
#choice: local, remote, pai
trainingServicePlatform: local
searchSpacePath: search_space.json
......@@ -11,10 +11,11 @@ useAnnotation: false
tuner:
#choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner
#SMAC (SMAC should be installed through nnictl)
builtinTunerName: TPE
builtinTunerName: GPTuner
classArgs:
#choice: maximize, minimize
optimize_mode: minimize
cold_start_num : 1
trial:
command: python3 main.py
codeDir: .
......
......@@ -11,10 +11,11 @@ useAnnotation: false
tuner:
#choice: TPE, Random, Anneal, Evolution, BatchTuner, MetisTuner, GPTuner
#SMAC (SMAC should be installed through nnictl)
builtinTunerName: TPE
builtinTunerName: GPTuner
classArgs:
#choice: maximize, minimize
optimize_mode: maximize
cold_start_num : 1
trial:
command: python3 mnist.py
codeDir: .
......
......@@ -40,7 +40,7 @@ def _match_val_type(vals, bounds):
vals_new.append(
min(bounds[i]['_value'], key=lambda x: abs(x - vals[i])))
elif _type in ['quniform', 'randint']:
vals_new.append(np.around(bounds[i]['_value']))
vals_new.append(np.around(vals[i]))
else:
vals_new.append(vals[i])
......
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