charlie880624/STDC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: print for clarified information in pytorch2onnx_kneron

Browse Source
This commit is contained in:
chingning.chen 2022-03-28 16:45:55 +08:00
parent caaa31c231
commit 4d7356fda7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/pytorch2onnx_kneron.py
View File

@ -227,9 +227,9 @@ def pytorch2onnx(model,
m.graph, i_n.name, norm_bn_bias, norm_bn_scale m.graph, i_n.name, norm_bn_bias, norm_bn_scale
) )
m = other.polish_model(m) 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) onnx.save(m, bn_outf)
print(f"ONNX for quantization saved to {bn_outf}") print(f"BN-Prepended ONNX saved to {bn_outf}")
return return