Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
cfd49acc
P
Paddle
项目概览
机器未来
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1
Issue
1
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
cfd49acc
编写于
8月 10, 2021
作者:
X
XGZhang
提交者:
GitHub
8月 10, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix a quantization bug (#34647)
上级
4f4662b0
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
3 addition
and
0 deletion
+3
-0
python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py
...d/contrib/slim/quantization/post_training_quantization.py
+1
-0
python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
...ddle/fluid/contrib/slim/quantization/quantization_pass.py
+2
-0
未找到文件。
python/paddle/fluid/contrib/slim/quantization/post_training_quantization.py
浏览文件 @
cfd49acc
...
@@ -578,6 +578,7 @@ class PostTrainingQuantization(object):
...
@@ -578,6 +578,7 @@ class PostTrainingQuantization(object):
var_tensor
=
_load_variable_data
(
self
.
_scope
,
var_name
)
var_tensor
=
_load_variable_data
(
self
.
_scope
,
var_name
)
var_tensor
=
var_tensor
.
flatten
()
var_tensor
=
var_tensor
.
flatten
()
abs_max_value
=
float
(
np
.
max
(
np
.
abs
(
var_tensor
)))
abs_max_value
=
float
(
np
.
max
(
np
.
abs
(
var_tensor
)))
abs_max_value
=
1e-8
if
abs_max_value
==
0.0
else
abs_max_value
s
=
0.3
s
=
0.3
if
var_name
not
in
self
.
_best_mse_loss
:
if
var_name
not
in
self
.
_best_mse_loss
:
self
.
_best_mse_loss
[
var_name
]
=
float
(
'inf'
)
self
.
_best_mse_loss
[
var_name
]
=
float
(
'inf'
)
...
...
python/paddle/fluid/contrib/slim/quantization/quantization_pass.py
浏览文件 @
cfd49acc
...
@@ -1312,6 +1312,7 @@ class QuantizationFreezePass(object):
...
@@ -1312,6 +1312,7 @@ class QuantizationFreezePass(object):
assert
self
.
_is_float
(
assert
self
.
_is_float
(
scale_v
),
'The scale of parameter %s is not a float.'
%
(
scale_v
),
'The scale of parameter %s is not a float.'
%
(
original_var_name
)
original_var_name
)
scale_v
=
1e-8
if
scale_v
==
0.0
else
scale_v
max_range
*=
param_range
/
scale_v
max_range
*=
param_range
/
scale_v
else
:
else
:
max_range
*=
act_range
max_range
*=
act_range
...
@@ -1413,6 +1414,7 @@ class QuantizationFreezePass(object):
...
@@ -1413,6 +1414,7 @@ class QuantizationFreezePass(object):
x
[:,
i
]
=
_clip
(
x
[:,
i
],
s
)
x
[:,
i
]
=
_clip
(
x
[:,
i
],
s
)
x
[:,
i
]
=
np
.
round
(
x
[:,
i
]
/
s
*
bnt
)
x
[:,
i
]
=
np
.
round
(
x
[:,
i
]
/
s
*
bnt
)
else
:
else
:
scale
=
1e-8
if
scale
==
0.0
else
scale
x
=
_clip
(
x
,
scale
)
x
=
_clip
(
x
,
scale
)
x
=
np
.
round
(
x
/
scale
*
bnt
)
x
=
np
.
round
(
x
/
scale
*
bnt
)
return
x
return
x
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录