File tree 1 file changed +3
-3
lines changed
python/paddle/fluid/contrib/slim/tests
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -42,10 +42,10 @@ if(LINUX AND WITH_MKLDNN)
42
42
inference_analysis_python_api_int8_test(test_slim_int8_googlenet ${INT8_GOOGLENET_MODEL_DIR} ${INT8_DATA_DIR} ${MKLDNN_INT8_TEST_FILE} )
43
43
44
44
# mobilenet int8
45
- set (INT8_MOBILENET_MODEL_DIR "${INT8_DATA_DIR} /mobilenet " )
45
+ set (INT8_MOBILENET_MODEL_DIR "${INT8_DATA_DIR} /mobilenetv1 " )
46
46
inference_analysis_python_api_int8_test(test_slim_int8_mobilenet ${INT8_MOBILENET_MODEL_DIR} ${INT8_DATA_DIR} ${MKLDNN_INT8_TEST_FILE} )
47
47
48
- # temporarily adding WITH_SLIM_MKLDNN_FULL_TEST FLAG for QA testing the following UTs locally,
48
+ # temporarily adding WITH_SLIM_MKLDNN_FULL_TEST FLAG for QA testing the following UTs locally,
49
49
# since the following UTs cost too much time on CI test.
50
50
if (WITH_SLIM_MKLDNN_FULL_TEST)
51
51
# resnet50 int8
@@ -70,7 +70,7 @@ if(LINUX AND WITH_MKLDNN)
70
70
endif ()
71
71
endif ()
72
72
73
- # Since test_mkldnn_int8_quantization_strategy only supports testing on Linux
73
+ # Since test_mkldnn_int8_quantization_strategy only supports testing on Linux
74
74
# with MKL-DNN, we remove it here for not repeating test, or not testing on other systems.
75
75
list (REMOVE_ITEM TEST_OPS test_mkldnn_int8_quantization_strategy)
76
76
You can’t perform that action at this time.
0 commit comments