return onnx.ModelProto

PINTO0309 · PINTO0309 · commit 5a9085dd82a7 · 2022-04-11T11:29:30.000+09:00
diff --git a/sne4onnx/onnx_network_extraction.py b/sne4onnx/onnx_network_extraction.py
@@ -114,7 +114,7 @@ def main():
     parser.add_argument(
         '--output_onnx_file_path',
         type=str,
-        default='',
+        default='extracted.onnx',
         help='Output onnx file path. If not specified, extracted.onnx is output.'
     )
     args = parser.parse_args()

Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ def main():`
`114`	`114`	`parser.add_argument(`
`115`	`115`	`'--output_onnx_file_path',`
`116`	`116`	`type=str,`
`117`		`- default='',`
	`117`	`+ default='extracted.onnx',`
`118`	`118`	`help='Output onnx file path. If not specified, extracted.onnx is output.'`
`119`	`119`	`)`
`120`	`120`	`args = parser.parse_args()`