File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
paddle/fluid/inference/tests/api Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -128,9 +128,9 @@ inference_analysis_api_test_with_fake_data(test_analyzer_resnet50
128
128
inference_analysis_api_test_with_fake_data (test_analyzer_mobilenet_depthwise_conv
129
129
"${INFERENCE_DEMO_INSTALL_DIR} /mobilenet_depthwise_conv" analyzer_resnet50_tester.cc "mobilenet_model.tar.gz" SERIAL )
130
130
131
- # bert, max_len=20
132
- set (BERT_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR} /bert20 " )
133
- download_model_and_data (${BERT_INSTALL_DIR} "bert_model .tar.gz" "bert_data_len20.txt.tar.gz" )
131
+ # bert, max_len=20, embedding_dim=128
132
+ set (BERT_INSTALL_DIR "${INFERENCE_DEMO_INSTALL_DIR} /bert_emb128 " )
133
+ download_model_and_data (${BERT_INSTALL_DIR} "bert_emb128_model .tar.gz" "bert_data_len20.txt.tar.gz" )
134
134
inference_analysis_api_test (test_analyzer_bert ${BERT_INSTALL_DIR} analyzer_bert_tester.cc SERIAL )
135
135
136
136
# anakin
You can’t perform that action at this time.
0 commit comments