Change help message for inputNames

This commit is contained in:
REDMOND\jingf 2021-09-09 16:18:42 -07:00
Родитель 6f5e0603e5
Коммит 90b031f21c
1 изменённых файлов: 1 добавлений и 3 удалений

Просмотреть файл

@ -7,7 +7,7 @@ def parse_args():
parser.add_argument('source', help='source model')
parser.add_argument('framework', help='source framework model comes from')
parser.add_argument('ONNXVersion', help='which ONNX Version to convert to')
parser.add_argument('inputNames', help='names of input nodes (optional)')
parser.add_argument('inputNames', help='names of input nodes')
parser.add_argument('outputNames', help='names of output nodes')
parser.add_argument('destination', help='destination ONNX model (ONNX or prototxt extension)')
parser.add_argument('--name', default='WimMLDashboardConvertedModel', help='(ONNX output only) model name')
@ -103,8 +103,6 @@ def convert_tensorflow_file(filename, opset, input_names, output_names):
import onnx
import tensorflow as tf
print(input_names)
graph_def = graph_pb2.GraphDef()
with open(filename, 'rb') as file:
graph_def.ParseFromString(file.read())