diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index 9790bd5b9688adc58263263d3601843b5efc8e77..bc9f713aeafb9977c60fe65bea56fbe2b395efd5 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,