saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix quantization tool bugs when model nodes have no name. (#3854)

Browse source 
Fix bugs when model nodes have no name.
This commit is contained in:
Zhang Lei 2020-05-12 20:38:26 -07:00 committed by GitHub
parent 9b5daa2039
commit eab61e87ce
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/quantization/calibrate.py
View file

@ -242,7 +242,9 @@ def main():
args = parser.parse_args()
calibrate_op_types = args.op_types.split(',')
black_nodes = args.black_nodes.split(',')
black_nodes = [x for x in black_nodes if x]
white_nodes = args.white_nodes.split(',')
white_nodes = [x for x in white_nodes if x]
model_path = args.model_path
output_model_path = args.output_model_path
images_folder = args.dataset_path