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

Fix datasize call in calibrate (#5110)

Browse source 
* Moving datasize to the interface.

* Reverting changes and adressing the comment
This commit is contained in:
Andrei Shadrikov 2020-09-15 01:06:23 +00:00 committed by GitHub
parent f7edf0aa57
commit 82b25e1731
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -126,7 +126,7 @@ class ONNXCalibrater:
intermediate_outputs.append(session.run(None, inputs))
node_output_names = [session.get_outputs()[i].name for i in range(len(intermediate_outputs[0]))]
output_dicts_list = [
dict(zip(node_output_names, intermediate_outputs[i])) for i in range(self.data_reader.datasize)
dict(zip(node_output_names, intermediate_output)) for intermediate_output in intermediate_outputs
]
#number of outputs in original model