From 4d7356fda7055df019ed06e2eec1717002e05afc Mon Sep 17 00:00:00 2001
From: "chingning.chen" <chingning.chen@kneron.us>
Date: Mon, 28 Mar 2022 16:45:55 +0800
Subject: [PATCH] chore: print for clarified information in pytorch2onnx_kneron

---
 tools/pytorch2onnx_kneron.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/pytorch2onnx_kneron.py b/tools/pytorch2onnx_kneron.py
index 4d89ad8..430bdc1 100644
--- a/tools/pytorch2onnx_kneron.py
+++ b/tools/pytorch2onnx_kneron.py
@@ -227,9 +227,9 @@ def pytorch2onnx(model,
             m.graph, i_n.name, norm_bn_bias, norm_bn_scale
         )
         m = other.polish_model(m)
-        bn_outf = os.path.splitext(output_file)[0] + "_kneron_optimized.onnx"
+        bn_outf = os.path.splitext(output_file)[0] + "_bn_prepended.onnx"
         onnx.save(m, bn_outf)
-        print(f"ONNX for quantization saved to {bn_outf}")
+        print(f"BN-Prepended ONNX saved to {bn_outf}")
 
     return