Skip to content

GitLab

Unverified Commit d750246c authored by QuanluZhang's avatar QuanluZhang Committed by GitHub
Browse files

fix convert from ss to pcs (#133)

parent 9cc9c68a
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
src/sdk/pynni/nni/smac_tuner/convert_ss_to_scenario.py
View file @ d750246c
...@@ -49,8 +49,8 @@ def generate_pcs(nni_search_space_content): ...@@ -49,8 +49,8 @@ def generate_pcs(nni_search_space_content):
if search_space[key]['_type'] == 'choice': if search_space[key]['_type'] == 'choice':
pcs_fd.write('%s categorical {%s} [%s]\n' % ( pcs_fd.write('%s categorical {%s} [%s]\n' % (
key, key,
str(search_space[key]['_value'])[1:-1], json.dumps(search_space[key]['_value'])[1:-1],
search_space[key]['_value'][0])) json.dumps(search_space[key]['_value'][0])))
elif search_space[key]['_type'] == 'randint': elif search_space[key]['_type'] == 'randint':
# TODO: support lower bound in randint # TODO: support lower bound in randint
pcs_fd.write('%s integer [0, %d] [%d]\n' % ( pcs_fd.write('%s integer [0, %d] [%d]\n' % (
...@@ -58,15 +58,15 @@ def generate_pcs(nni_search_space_content): ...@@ -58,15 +58,15 @@ def generate_pcs(nni_search_space_content):
search_space[key]['_value'][0], search_space[key]['_value'][0],
search_space[key]['_value'][0])) search_space[key]['_value'][0]))
elif search_space[key]['_type'] == 'uniform': elif search_space[key]['_type'] == 'uniform':
pcs_fd.write('%s real %s [%f]\n' % ( pcs_fd.write('%s real %s [%s]\n' % (
key, key,
search_space[key]['_value'], json.dumps(search_space[key]['_value']),
search_space[key]['_value'][0])) json.dumps(search_space[key]['_value'][0])))
elif search_space[key]['_type'] == 'loguniform': elif search_space[key]['_type'] == 'loguniform':
pcs_fd.write('%s real %s [%f] log\n' % ( pcs_fd.write('%s real %s [%s] log\n' % (
key, key,
search_space[key]['_value'], json.dumps(search_space[key]['_value']),
search_space[key]['_value'][0])) json.dumps(search_space[key]['_value'][0])))
elif search_space[key]['_type'] == 'quniform' \ elif search_space[key]['_type'] == 'quniform' \
and search_space[key]['_value'][2] == 1: and search_space[key]['_value'][2] == 1:
pcs_fd.write('%s integer [%d, %d] [%d]\n' % ( pcs_fd.write('%s integer [%d, %d] [%d]\n' % (
......
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