Skip to content

Commit cd6d244

Browse files
authored
Fix onnxrt calibration issue (#1666)
Signed-off-by: Mengni Wang <[email protected]>
1 parent 01d966b commit cd6d244

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

neural_compressor/adaptor/ox_utils/calibration.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -440,8 +440,9 @@ def get_weight_tensors_calib_range(self):
440440
),
441441
)
442442
calibrator = CALIBRATOR["minmax"]() # use minmax method to calibrate initializer tensors
443-
calibrator.collect(initializer_tensor)
444-
weight_tensors_calib_range[initializer_tensor_name] = [list(calibrator.calib_range)]
443+
if initializer_tensor.flatten().size > 0:
444+
calibrator.collect(initializer_tensor)
445+
weight_tensors_calib_range[initializer_tensor_name] = [list(calibrator.calib_range)]
445446
calibrator.clear()
446447
del calibrator
447448
return weight_tensors_calib_range

0 commit comments

Comments
 (0)