提交 708bec2e 编写于 作者: T tensor-tang

add test

上级 d13dd3b6
......@@ -200,6 +200,14 @@ TEST(inference, nlp) {
LOG(INFO) << "Total infer time: " << (stop_ms - start_ms) / 1000.0 / 60
<< " min, avg time per seq: "
<< (stop_ms - start_ms) / datasets.size() << " ms";
{ // just for test
auto* scope = new paddle::framework::Scope();
paddle::framework::LoDTensor outtensor;
TestInference<paddle::platform::CPUPlace, false, true>(
dirname, {&(datasets[0])}, {&outtensor}, FLAGS_repeat, model_combined,
false);
delete scope;
}
}
delete scope;
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册