未验证 提交 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:
- !Permute
to_bgr: false
channel_first: True
batch_size: 8
batch_size: 1
drop_empty: false
worker_num: 4
bufsize: 2
......
......@@ -113,9 +113,8 @@ endif()
if (NOT WIN32)
if (WITH_TENSORRT AND WITH_GPU)
include_directories("${TENSORRT_INC_DIR}/include")
#link_directories("${TENSORRT_LIB_DIR}/lib")
link_directories("${TENSORRT_LIB_DIR}/")
include_directories("${TENSORRT_INC_DIR}")
link_directories("${TENSORRT_LIB_DIR}")
endif()
endif(NOT WIN32)
......@@ -149,7 +148,11 @@ if(WITH_MKL)
endif ()
endif()
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()
if (WIN32)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册