use cls_batch_num replace rec_batch_num (#1298)

2026-09-24 23:33:08 +08:00 · 2020-12-02 12:22:41 +08:00
parent 9b2c0e4838
commit 2985e7b897
1 changed files with 1 additions and 1 deletions
@@ -37,7 +37,7 @@ logger = get_logger()
 class TextClassifier(object):
    def __init__(self, args):
        self.cls_image_shape = [int(v) for v in args.cls_image_shape.split(",")]
-        self.cls_batch_num = args.rec_batch_num
+        self.cls_batch_num = args.cls_batch_num
        self.cls_thresh = args.cls_thresh
        self.use_zero_copy_run = args.use_zero_copy_run
        postprocess_params = {