运行C++ LIC2020关系抽取 demo¶
该工程是2020语言与智能技术竞赛:关系抽取任务竞赛提供的基线模型的c++预测demo,基线模型的训练相关信息可参考LIC2020-关系抽取任务基线系统。
一:获取基线模型¶
点击链接下载模型,如果你想获取更多的模型训练信息,请访问LIC2020-关系抽取任务基线系统。
二:样例编译¶
文件demo.cc
为预测的样例程序(程序中的输入为固定值,如果您有其他方式进行数据读取的需求,需要对程序进行一定的修改)。
文件CMakeLists.txt
为编译构建文件。
脚本run_impl.sh
包含了第三方库、预编译库的信息配置。
编译demo样例,我们首先需要对脚本run_impl.sh
文件中的配置进行修改。
1)修改run_impl.sh
打开run_impl.sh
,我们对以下的几处信息进行修改:
# 根据预编译库中的version.txt信息判断是否将以下三个标记打开
WITH_MKL=ON
WITH_GPU=ON
USE_TENSORRT=OFF
# 配置预测库的根目录
LIB_DIR=${YOUR_LIB_DIR}/fluid_inference_install_dir
# 如果上述的WITH_GPU 或 USE_TENSORRT设为ON,请设置对应的CUDA, CUDNN, TENSORRT的路径。
CUDNN_LIB=/usr/local/cudnn/lib64
CUDA_LIB=/usr/local/cuda/lib64
# TENSORRT_ROOT=/usr/local/TensorRT-6.0.1.5
运行 sh run_impl.sh
, 会在目录下产生build目录。
2) 运行样例
# 进入build目录
cd build
# 运行样例
./demo -model_file ${LIC_MODEL_PATH}/model --params_file ${LIC_MODEL_PATH}/params
运行结束后,程序会将模型输出个数打印到屏幕,说明运行成功。