IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: check batch bias shape equal one in fbc (#9907)

Browse Source
This commit is contained in:
Indira Salyahova 2022-01-26 14:37:43 +03:00 committed by GitHub
parent 251897a91a
commit cf49cd6ce8
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
tools/pot/openvino/tools/pot/algorithms/quantization/fast_bias_correction/algorithm.py
View File

@ -215,4 +215,6 @@ class FastBiasCorrection(Algorithm):
_, inputs_shape = self._engine.predict(calculate_input_shape, sampler)
for node_name, shape_node in inputs_shape.items():
inputs_shape[node_name] = shape_node['shape_node'][0]
if len(inputs_shape[node_name]) > 1:
inputs_shape[node_name] = (1, *inputs_shape[node_name][1:])
return inputs_shape