From 203c90ed549e74de58b99fbdb548fac7321f4504 Mon Sep 17 00:00:00 2001 From: jack <136876878@qq.com> Date: Thu, 2 Jul 2020 10:00:49 +0800 Subject: [PATCH] remove batch_size arg --- deploy/cpp/demo/classifier.cpp | 3 +-- deploy/cpp/demo/detector.cpp | 3 +-- deploy/cpp/demo/segmenter.cpp | 3 +-- 3 files changed, 3 insertions(+), 6 deletions(-) diff --git a/deploy/cpp/demo/classifier.cpp b/deploy/cpp/demo/classifier.cpp index e616d0e..8b78d7e 100644 --- a/deploy/cpp/demo/classifier.cpp +++ b/deploy/cpp/demo/classifier.cpp @@ -57,8 +57,7 @@ int main(int argc, char** argv) { FLAGS_use_gpu, FLAGS_use_trt, FLAGS_gpu_id, - FLAGS_key, - FLAGS_batch_size); + FLAGS_key); // 进行预测 double total_running_time_s = 0.0; diff --git a/deploy/cpp/demo/detector.cpp b/deploy/cpp/demo/detector.cpp index 07bb766..5b4e3a2 100644 --- a/deploy/cpp/demo/detector.cpp +++ b/deploy/cpp/demo/detector.cpp @@ -62,8 +62,7 @@ int main(int argc, char** argv) { FLAGS_use_gpu, FLAGS_use_trt, FLAGS_gpu_id, - FLAGS_key, - FLAGS_batch_size); + FLAGS_key); double total_running_time_s = 0.0; double total_imread_time_s = 0.0; diff --git a/deploy/cpp/demo/segmenter.cpp b/deploy/cpp/demo/segmenter.cpp index b9a8188..7dd48e5 100644 --- a/deploy/cpp/demo/segmenter.cpp +++ b/deploy/cpp/demo/segmenter.cpp @@ -59,8 +59,7 @@ int main(int argc, char** argv) { FLAGS_use_gpu, FLAGS_use_trt, FLAGS_gpu_id, - FLAGS_key, - FLAGS_batch_size); + FLAGS_key); double total_running_time_s = 0.0; double total_imread_time_s = 0.0; -- GitLab