Skip to content
This repository has been archived by the owner on Sep 18, 2024. It is now read-only.

Fix includeIntermediateResults field in config_schema #1326

Merged
merged 9 commits into from
Jul 25, 2019
Merged

Fix includeIntermediateResults field in config_schema #1326

Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
9fb25cc
Merge pull request #189 from microsoft/master
SparkSnail Jul 17, 2019
cb25d66
Merge pull request #190 from microsoft/master
SparkSnail Jul 18, 2019
1bd2012
Merge pull request #191 from microsoft/master
SparkSnail Jul 18, 2019
55a5628
fix includeintermediateresult
SparkSnail Jul 18, 2019
e39485a
test pr tigger
SparkSnail Jul 18, 2019
c76068f
Merge pull request #192 from microsoft/master
SparkSnail Jul 23, 2019
803b6e4
Merge pull request #193 from microsoft/master
SparkSnail Jul 24, 2019
5830533
Merge branch 'master' of https://github.com/SparkSnail/nni into fix-n…
SparkSnail Jul 24, 2019
bcd591f
add evolution field
SparkSnail Jul 24, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions tools/nni_cmd/config_schema.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -84,10 +84,12 @@ def setPathCheck(key):
'optimize_mode': setChoice('optimize_mode', 'maximize', 'minimize'),
Optional('population_size'): setNumberRange('population_size', int, 0, 99999),
},
Optional('includeIntermediateResults'): setType('includeIntermediateResults', bool),
Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999),
},
('BatchTuner', 'GridSearch', 'Random'): {
'builtinTunerName': setChoice('builtinTunerName', 'BatchTuner', 'GridSearch', 'Random'),
Optional('includeIntermediateResults'): setType('includeIntermediateResults', bool),
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think we could hold a discussion about refactoring these logic.

Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999),
},
'NetworkMorphism': {
Expand All @@ -99,6 +101,7 @@ def setPathCheck(key):
Optional('input_channel'): setType('input_channel', int),
Optional('n_output_node'): setType('n_output_node', int),
},
Optional('includeIntermediateResults'): setType('includeIntermediateResults', bool),
Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999),
},
'MetisTuner': {
Expand All @@ -110,6 +113,7 @@ def setPathCheck(key):
Optional('selection_num_starting_points'): setType('selection_num_starting_points', int),
Optional('cold_start_num'): setType('cold_start_num', int),
},
Optional('includeIntermediateResults'): setType('includeIntermediateResults', bool),
Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999),
},
'GPTuner': {
Expand All @@ -125,13 +129,15 @@ def setPathCheck(key):
Optional('selection_num_warm_up'): setType('selection_num_warm_up', int),
Optional('selection_num_starting_points'): setType('selection_num_starting_points', int),
},
Optional('includeIntermediateResults'): setType('includeIntermediateResults', bool),
Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999),
},
'customized': {
'codeDir': setPathCheck('codeDir'),
'classFileName': setType('classFileName', str),
'className': setType('className', str),
Optional('classArgs'): dict,
Optional('includeIntermediateResults'): setType('includeIntermediateResults', bool),
Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999),
}
}
Expand Down