Skip to content

GitLab

Commit 3ff007de authored by Lee's avatar Lee Committed by chicm-ms
Browse files

Make population_size optional (#1334)

parent aa8b09e6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/nni_cmd/config_schema.py
View file @ 3ff007de
...@@ -82,7 +82,7 @@ tuner_schema_dict = { ...@@ -82,7 +82,7 @@ tuner_schema_dict = {
'builtinTunerName': setChoice('builtinTunerName', 'Evolution'), 'builtinTunerName': setChoice('builtinTunerName', 'Evolution'),
Optional('classArgs'): { Optional('classArgs'): {
'optimize_mode': setChoice('optimize_mode', 'maximize', 'minimize'), 'optimize_mode': setChoice('optimize_mode', 'maximize', 'minimize'),
'population_size': setNumberRange('population_size', int, 0, 99999), Optional('population_size'): setNumberRange('population_size', int, 0, 99999),
}, },
Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999), Optional('gpuNum'): setNumberRange('gpuNum', int, 0, 99999),
}, },
......
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