From 7c9a5bd68c76cb1bdbbd26f39fc18700596c7a20 Mon Sep 17 00:00:00 2001
From: tink2123 <y_tink@163.com>
Date: Thu, 8 Jul 2021 16:34:05 +0800
Subject: [PATCH] fix args

---
 tools/infer/predict_rec.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py
index 9790bd5b..bc9f713a 100755
--- a/tools/infer/predict_rec.py
+++ b/tools/infer/predict_rec.py
@@ -71,9 +71,9 @@ class TextRecognizer(object):
             self.autolog = auto_log.AutoLogger(
                 model_name="rec",
                 model_precision=args.precision,
-                batch_size=6,
+                batch_size=args.rec_batch_num,
                 data_shape="dynamic",
-                save_path="./output/auto_log.lpg",
+                save_path=args.save_log_path,
                 inference_config=self.config,
                 pids=pid,
                 process_name=None,
-- 
GitLab