Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
吃玉米的猫
Paddle
提交
cd74b207
P
Paddle
项目概览
吃玉米的猫
/
Paddle
与 Fork 源项目一致
Fork自
PaddlePaddle / Paddle
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
cd74b207
编写于
4月 02, 2021
作者:
C
cc
提交者:
GitHub
4月 02, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add more ops to calculate output scales (#32036)
上级
ed49b418
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
13 addition
and
23 deletion
+13
-23
python/paddle/fluid/contrib/slim/quantization/imperative/qat.py
.../paddle/fluid/contrib/slim/quantization/imperative/qat.py
+1
-1
python/paddle/fluid/contrib/slim/quantization/imperative/utils.py
...addle/fluid/contrib/slim/quantization/imperative/utils.py
+12
-22
未找到文件。
python/paddle/fluid/contrib/slim/quantization/imperative/qat.py
浏览文件 @
cd74b207
...
...
@@ -468,7 +468,7 @@ class ImperativeQuantizeOutputs(object):
"""
Whether the layer needs to calculate output scales.
"""
return
isinstance
(
layer
,
tuple
(
utils
.
quant_output_layers_map
.
values
())
)
\
return
isinstance
(
layer
,
utils
.
quant_output_layers
)
\
or
(
'quantized'
in
layer
.
full_name
()
and
\
'quantized_noweight'
not
in
layer
.
full_name
())
...
...
python/paddle/fluid/contrib/slim/quantization/imperative/utils.py
浏览文件 @
cd74b207
...
...
@@ -43,28 +43,18 @@ fake_quantize_dequantize_types = [
"fake_quantize_dequantize_moving_average_abs_max"
]
quant_output_layers_map
=
{
'Conv2D'
:
paddle
.
nn
.
Conv2D
,
'Conv2DTranspose'
:
paddle
.
nn
.
Conv2DTranspose
,
'Linear'
:
paddle
.
nn
.
Linear
,
'AdaptiveAvgPool2D'
:
paddle
.
nn
.
AdaptiveAvgPool2D
,
'AdaptiveMaxPool2D'
:
paddle
.
nn
.
AdaptiveMaxPool2D
,
'AvgPool2D'
:
paddle
.
nn
.
AvgPool2D
,
'MaxPool2D'
:
paddle
.
nn
.
MaxPool2D
,
'BatchNorm'
:
paddle
.
nn
.
BatchNorm
,
'BatchNorm2D'
:
paddle
.
nn
.
BatchNorm2D
,
'SyncBatchNorm'
:
paddle
.
nn
.
SyncBatchNorm
,
'ELU'
:
paddle
.
nn
.
ELU
,
'GELU'
:
paddle
.
nn
.
GELU
,
'LeakyReLU'
:
paddle
.
nn
.
LeakyReLU
,
'PReLU'
:
paddle
.
nn
.
PReLU
,
'ReLU'
:
paddle
.
nn
.
ReLU
,
'ReLU6'
:
paddle
.
nn
.
ReLU6
,
'Sigmoid'
:
paddle
.
nn
.
Sigmoid
,
'Softmax'
:
paddle
.
nn
.
Softmax
,
'Tanh'
:
paddle
.
nn
.
Tanh
,
'Swish'
:
paddle
.
nn
.
Swish
,
}
quant_output_layers
=
(
paddle
.
nn
.
Conv2D
,
paddle
.
nn
.
Conv2DTranspose
,
paddle
.
nn
.
Linear
,
paddle
.
nn
.
AdaptiveAvgPool2D
,
paddle
.
nn
.
AdaptiveMaxPool2D
,
paddle
.
nn
.
AvgPool2D
,
paddle
.
nn
.
MaxPool2D
,
paddle
.
nn
.
BatchNorm
,
paddle
.
nn
.
BatchNorm2D
,
paddle
.
nn
.
LayerNorm
,
paddle
.
nn
.
SyncBatchNorm
,
paddle
.
nn
.
ELU
,
paddle
.
nn
.
GELU
,
paddle
.
nn
.
Hardshrink
,
paddle
.
nn
.
Hardsigmoid
,
paddle
.
nn
.
Hardswish
,
paddle
.
nn
.
Hardtanh
,
paddle
.
nn
.
LeakyReLU
,
paddle
.
nn
.
LogSigmoid
,
paddle
.
nn
.
LogSoftmax
,
paddle
.
nn
.
Maxout
,
paddle
.
nn
.
PReLU
,
paddle
.
nn
.
ReLU
,
paddle
.
nn
.
ReLU6
,
paddle
.
nn
.
SELU
,
paddle
.
nn
.
Sigmoid
,
paddle
.
nn
.
Softmax
,
paddle
.
nn
.
Softplus
,
paddle
.
nn
.
Softshrink
,
paddle
.
nn
.
Softsign
,
paddle
.
nn
.
Swish
,
paddle
.
nn
.
Tanh
,
paddle
.
nn
.
Tanhshrink
,
paddle
.
nn
.
ThresholdedReLU
,
paddle
.
nn
.
Upsample
)
weight_op_types
=
[
"conv2d"
,
"depthwise_conv2d"
,
"matmul"
,
"conv2d_transpose"
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录