Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
机器未来
Paddle
提交
5c61eeef
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看板
未验证
提交
5c61eeef
编写于
11月 30, 2020
作者:
J
joanna.wozna.intel
提交者:
GitHub
11月 30, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Enable all image classification models (#29155)
上级
d8eef4e4
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
14 addition
and
7 deletion
+14
-7
paddle/fluid/inference/tests/api/CMakeLists.txt
paddle/fluid/inference/tests/api/CMakeLists.txt
+9
-0
paddle/fluid/inference/tests/api/analyzer_bfloat16_image_classification_tester.cc
...ests/api/analyzer_bfloat16_image_classification_tester.cc
+5
-7
未找到文件。
paddle/fluid/inference/tests/api/CMakeLists.txt
浏览文件 @
5c61eeef
...
@@ -375,6 +375,15 @@ if(WITH_MKLDNN)
...
@@ -375,6 +375,15 @@ if(WITH_MKLDNN)
# resnet50 bfloat16
# resnet50 bfloat16
inference_analysis_api_bfloat16_test_run
(
test_analyzer_bfloat16_resnet50
${
BF16_IMG_CLASS_TEST_APP
}
${
INT8_RESNET50_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
inference_analysis_api_bfloat16_test_run
(
test_analyzer_bfloat16_resnet50
${
BF16_IMG_CLASS_TEST_APP
}
${
INT8_RESNET50_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
# googlenet bfloat16
inference_analysis_api_bfloat16_test_run
(
test_analyzer_bfloat16_googlenet
${
BF16_IMG_CLASS_TEST_APP
}
${
INT8_GOOGLENET_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
# mobilenetv1 bfloat16
inference_analysis_api_bfloat16_test_run
(
test_analyzer_bfloat16_mobilenetv1
${
BF16_IMG_CLASS_TEST_APP
}
${
INT8_MOBILENETV1_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
# mobilenetv2 bfloat16
inference_analysis_api_bfloat16_test_run
(
test_analyzer_bfloat16_mobilenetv2
${
BF16_IMG_CLASS_TEST_APP
}
${
INT8_MOBILENETV2_MODEL_DIR
}
${
IMAGENET_DATA_PATH
}
)
### Object detection models
### Object detection models
set
(
PASCALVOC_DATA_PATH
"
${
INT8_DATA_DIR
}
/pascalvoc_val_head_300.bin"
)
set
(
PASCALVOC_DATA_PATH
"
${
INT8_DATA_DIR
}
/pascalvoc_val_head_300.bin"
)
...
...
paddle/fluid/inference/tests/api/analyzer_bfloat16_image_classification_tester.cc
浏览文件 @
5c61eeef
...
@@ -28,20 +28,18 @@ void SetConfig(AnalysisConfig *cfg) {
...
@@ -28,20 +28,18 @@ void SetConfig(AnalysisConfig *cfg) {
cfg
->
EnableMKLDNN
();
cfg
->
EnableMKLDNN
();
}
}
TEST
(
Analyzer_
int8
_image_classification
,
bfloat16
)
{
TEST
(
Analyzer_
bfloat16
_image_classification
,
bfloat16
)
{
AnalysisConfig
cfg
;
AnalysisConfig
cfg
;
SetConfig
(
&
cfg
);
SetConfig
(
&
cfg
);
AnalysisConfig
q
_cfg
;
AnalysisConfig
b
_cfg
;
SetConfig
(
&
q
_cfg
);
SetConfig
(
&
b
_cfg
);
// read data from file and prepare batches with test data
// read data from file and prepare batches with test data
std
::
vector
<
std
::
vector
<
PaddleTensor
>>
input_slots_all
;
std
::
vector
<
std
::
vector
<
PaddleTensor
>>
input_slots_all
;
SetInputs
(
&
input_slots_all
);
SetInputs
(
&
input_slots_all
);
q_cfg
.
SwitchIrDebug
();
b_cfg
.
EnableMkldnnBfloat16
();
q_cfg
.
EnableMkldnnBfloat16
();
CompareBFloat16AndAnalysis
(
&
cfg
,
&
b_cfg
,
input_slots_all
);
q_cfg
.
SetBfloat16Op
({
"conv2d"
});
CompareBFloat16AndAnalysis
(
&
cfg
,
&
q_cfg
,
input_slots_all
);
}
}
}
// namespace analysis
}
// namespace analysis
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录