未验证 提交 857b17b8 编写于 作者: C cnn 提交者: GitHub

set batch_size=1 when eval, and fix bug of cpp build on winodows (#1822)

上级 aeda569d
...@@ -144,7 +144,7 @@ EvalReader: ...@@ -144,7 +144,7 @@ EvalReader:
- !Permute - !Permute
to_bgr: false to_bgr: false
channel_first: True channel_first: True
batch_size: 8 batch_size: 1
drop_empty: false drop_empty: false
worker_num: 4 worker_num: 4
bufsize: 2 bufsize: 2
......
...@@ -113,9 +113,8 @@ endif() ...@@ -113,9 +113,8 @@ endif()
if (NOT WIN32) if (NOT WIN32)
if (WITH_TENSORRT AND WITH_GPU) if (WITH_TENSORRT AND WITH_GPU)
include_directories("${TENSORRT_INC_DIR}/include") include_directories("${TENSORRT_INC_DIR}")
#link_directories("${TENSORRT_LIB_DIR}/lib") link_directories("${TENSORRT_LIB_DIR}")
link_directories("${TENSORRT_LIB_DIR}/")
endif() endif()
endif(NOT WIN32) endif(NOT WIN32)
...@@ -149,7 +148,11 @@ if(WITH_MKL) ...@@ -149,7 +148,11 @@ if(WITH_MKL)
endif () endif ()
endif() endif()
else() else()
set(MATH_LIB ${PADDLE_DIR}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX}) if (WIN32)
set(MATH_LIB ${PADDLE_DIR}/third_party/install/openblas/lib/openblas${CMAKE_STATIC_LIBRARY_SUFFIX})
else()
set(MATH_LIB ${PADDLE_DIR}/third_party/install/openblas/lib/libopenblas${CMAKE_STATIC_LIBRARY_SUFFIX})
endif()
endif() endif()
if (WIN32) if (WIN32)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册