Skip to content

Commit f351ceb

Browse files
authored
Merge pull request #13345 from Superjomn/bugfix/lac_test
fix ner_test when bs>1
2 parents 1b676be + 8e0fe03 commit f351ceb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddle/fluid/inference/tests/api/analyzer_ner_tester.cc

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,8 +144,9 @@ void TestChineseNERPrediction(bool use_analysis) {
144144
size_t num_samples;
145145
for (int i = 0; i < FLAGS_repeat; i++) {
146146
DataRecord data(FLAGS_infer_data, FLAGS_batch_size);
147+
// Just one batch, the num_samples remains the same.
147148
num_samples = data.num_samples;
148-
for (size_t bid = 0; bid < num_samples; ++bid) {
149+
for (size_t bid = 0; bid < num_samples / FLAGS_batch_size; ++bid) {
149150
PrepareInputs(&input_slots, &data, FLAGS_batch_size);
150151
timer.tic();
151152
predictor->Run(input_slots, &outputs);

0 commit comments

Comments
 (0)