Skip to content

Commit

Permalink
[SAMPLES] Get the correct batch size when running with pre-compiled m…
Browse files Browse the repository at this point in the history
…odels (openvinotoolkit#24507)

### Details:
 - *get the correct batch size when running with pre-compiled models*

### Tickets:
 - *CVS-140703*
  • Loading branch information
pereanub authored May 15, 2024
1 parent 058aed8 commit 3302535
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions samples/cpp/benchmark_app/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -792,6 +792,11 @@ int main(int argc, char* argv[]) {
FLAGS_scale_values,
FLAGS_mean_values,
compiledModel.inputs());

batchSize = get_batch_size(app_inputs_info.at(0));
warn_if_no_batch(app_inputs_info.at(0));
slog::info << "Model batch size: " << batchSize << slog::endl;

if (batchSize == 0) {
batchSize = 1;
}
Expand Down

0 comments on commit 3302535

Please sign in to comment.