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

fix: don't pass parametr inplace_statistic for weights (#10593)

Browse Source
This commit is contained in:
Indira Salyahova 2022-02-22 16:53:53 +03:00 committed by GitHub
parent 5bb8f77c3f
commit aced89a655
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/pot/openvino/tools/pot/algorithms/quantization/utils.py
View File

@ -293,7 +293,8 @@ def get_tensor_statistics(range_estimator_config, for_weights, **kwargs):
stat_mod_name = get_stat_name_by_config(range_estimator_config, stats_name) stat_mod_name = get_stat_name_by_config(range_estimator_config, stats_name)
if fn_type in ['quantile', 'abs_quantile']: if fn_type in ['quantile', 'abs_quantile']:
q_value = range_estimator_config[stats_name]['outlier_prob'] q_value = range_estimator_config[stats_name]['outlier_prob']
ts_args['inplace_statistics'] = False if not for_weights:
ts_args['inplace_statistics'] = False
if stats_name == 'max': if stats_name == 'max':
q_value = 1 - q_value q_value = 1 - q_value
ts_args.update({'q': q_value}) ts_args.update({'q': q_value})