未验证 提交 5fcd8061 编写于 作者: T Tomasz Socha 提交者: GitHub

Enable bert model on CPU (#43244)

* Enable bert model on CPU

* Style
上级 562b184c
......@@ -869,10 +869,8 @@ endif()
set(BERT_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/bert_emb128")
download_model_and_data_without_verify(
${BERT_INSTALL_DIR} "bert_emb128_model.tar.gz" "bert_data_len20.txt.tar.gz")
if(WITH_GPU)
inference_analysis_api_test(test_analyzer_bert ${BERT_INSTALL_DIR}
analyzer_bert_tester.cc)
endif()
inference_analysis_api_test(test_analyzer_bert ${BERT_INSTALL_DIR}
analyzer_bert_tester.cc)
# multiple models prediction
set(MMP_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR}/multi_model_prediction")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册