Add dump binary config

feature/design_of_v2_layer_converter
Yu Yang 8 years ago
parent 987a908f29
commit 3b5bed68d0

@ -20,6 +20,7 @@ __all__ = []
if __name__ == '__main__':
whole_conf = False
binary = False
if len(sys.argv) == 2:
conf = parse_config(sys.argv[1], '')
elif len(sys.argv) == 3:
@ -28,6 +29,8 @@ if __name__ == '__main__':
conf = parse_config(sys.argv[1], sys.argv[2])
if sys.argv[3] == '--whole':
whole_conf = True
elif sys.argv[3] == '--binary':
binary = True
else:
raise RuntimeError()
@ -36,4 +39,7 @@ if __name__ == '__main__':
if whole_conf:
print conf
else:
print conf.model_config
if binary:
sys.stdout.write(conf.SerializeToString())
else:
print conf.model_config

Loading…
Cancel
Save