瀏覽代碼

remove batch_size arg

jack 5 年之前
父節點
當前提交
203c90ed54
共有 3 個文件被更改,包括 3 次插入6 次删除
  1. 1 2
      deploy/cpp/demo/classifier.cpp
  2. 1 2
      deploy/cpp/demo/detector.cpp
  3. 1 2
      deploy/cpp/demo/segmenter.cpp

+ 1 - 2
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;

+ 1 - 2
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;

+ 1 - 2
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;