Change constant to variable in ocr model(#3377)

ee882b53 · whs · GitHub · ad1a917f · ee882b53
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleCV/ocr_recognition/attention_model.py PaddleCV/ocr_recognition/attention_model.py +1 -1

未找到文件。
--- a/PaddleCV/ocr_recognition/attention_model.py
+++ b/PaddleCV/ocr_recognition/attention_model.py
@@ -325,7 +325,7 @@ def attention_infer(images, num_classes, use_cudnn=True):
            topk_indices,
            accu_scores,
            beam_size,
-            1,  # end_id
+            eos,  # end_id
            #level=0
        )