diff --git a/converters/nnet2onnx.py b/converters/nnet2onnx.py index 0ec800a..bcf0dbb 100644 --- a/converters/nnet2onnx.py +++ b/converters/nnet2onnx.py @@ -78,7 +78,7 @@ def nnet2onnx(nnetFile, onnxFile="", outputVar = "y_out", inputVar="X", normaliz if len(sys.argv)>2: onnxFile = sys.argv[2] if len(sys.argv)>3: - outputName = argv[3] + outputName = sys.argv[3] nnet2onnx(nnetFile,onnxFile,outputName) else: nnet2onnx(nnetFile,onnxFile) else: nnet2onnx(nnetFile)