Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
bd955c75
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
bd955c75
编写于
8月 24, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
8月 24, 2020
浏览文件
操作
浏览文件
下载
差异文件
!5067 fix post training quant log
Merge pull request !5067 from xutianchun/quant_0824
上级
0f69259a
d861a619
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
10 addition
and
3 deletion
+10
-3
mindspore/lite/tools/anf_exporter/anf_exporter.cc
mindspore/lite/tools/anf_exporter/anf_exporter.cc
+8
-1
mindspore/lite/tools/converter/quantizer/quantize_util.cc
mindspore/lite/tools/converter/quantizer/quantize_util.cc
+2
-2
未找到文件。
mindspore/lite/tools/anf_exporter/anf_exporter.cc
浏览文件 @
bd955c75
...
...
@@ -69,6 +69,10 @@ int AnfExporter::ConvertQuantParam(const std::unique_ptr<schema::MetaGraphT> &me
// activation
auto
input_quant_params
=
primitive
->
GetInputQuantParams
();
auto
node_type
=
(
schema
::
PrimitiveType
)
primitive
->
Type
();
if
(
input_quant_params
.
empty
())
{
MS_LOG
(
ERROR
)
<<
"node: "
<<
dst_node
->
name
<<
" input quant params is empty"
;
return
RET_ERROR
;
}
for
(
size_t
i
=
0
;
i
<
input_quant_params
.
size
();
i
++
)
{
if
(
i
>=
dst_node
->
inputIndex
.
size
())
{
MS_LOG
(
ERROR
)
<<
"node: "
<<
dst_node
->
name
<<
" input has "
<<
input_quant_params
.
size
()
...
...
@@ -93,7 +97,10 @@ int AnfExporter::ConvertQuantParam(const std::unique_ptr<schema::MetaGraphT> &me
auto
tensor_output
=
meta_graph
->
allTensors
[
output_index
].
get
();
auto
output_quant_params
=
primitive
->
GetOutputQuantParams
();
if
(
output_quant_params
.
empty
())
{
MS_LOG
(
WARNING
)
<<
"node: "
<<
dst_node
->
name
<<
" output quant params is empty"
;
if
(
node_type
!=
schema
::
PrimitiveType_QuantDTypeCast
)
{
MS_LOG
(
ERROR
)
<<
"node: "
<<
dst_node
->
name
<<
" output quant params is empty"
;
return
RET_ERROR
;
}
}
else
{
for
(
auto
output_quant_param
:
output_quant_params
[
0
])
{
if
(
tensor_output
->
quantParams
.
empty
())
{
...
...
mindspore/lite/tools/converter/quantizer/quantize_util.cc
浏览文件 @
bd955c75
...
...
@@ -168,11 +168,11 @@ STATUS CalQuantizationParams(schema::QuantParamT *quantParam, double mMin, doubl
int
quant_max
,
int
quant_min
,
int
num_bits
)
{
MS_ASSERT
(
quantParam
!=
nullptr
);
if
(
mMin
>
0.0
f
)
{
MS_LOG
(
ERROR
)
<<
"min "
<<
mMin
<<
" is bigger then 0, set to 0, this may course low precision"
;
MS_LOG
(
DEBUG
)
<<
"min "
<<
mMin
<<
" is bigger then 0, set to 0, this may course low precision"
;
mMin
=
0.0
f
;
}
if
(
mMax
<
0.0
f
)
{
MS_LOG
(
ERROR
)
<<
"mMax "
<<
mMax
<<
" is smaller than 0, set to 0, this may course low precision"
;
MS_LOG
(
DEBUG
)
<<
"mMax "
<<
mMax
<<
" is smaller than 0, set to 0, this may course low precision"
;
mMax
=
0.0
f
;
}
if
(
mMin
>
mMax
)
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录