Change help message for inputNames
This commit is contained in:
Родитель
6f5e0603e5
Коммит
90b031f21c
|
@ -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())
|
||||
|
|
Загрузка…
Ссылка в новой задаче