From b68c895827641b3459069fe804546407fd6c8711 Mon Sep 17 00:00:00 2001 From: andyjpaddle Date: Tue, 7 Sep 2021 07:39:24 +0000 Subject: [PATCH] fix code style --- ppocr/modeling/heads/rec_sar_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/modeling/heads/rec_sar_head.py b/ppocr/modeling/heads/rec_sar_head.py index 3c131c8b..647f5820 100644 --- a/ppocr/modeling/heads/rec_sar_head.py +++ b/ppocr/modeling/heads/rec_sar_head.py @@ -195,7 +195,7 @@ class ParallelSARDecoder(BaseDecoder): # Prediction layer self.pred_dropout = nn.Dropout(pred_dropout) - pred_num_classes = num_classes - 1 + pred_num_classes = self.num_classes - 1 if pred_concat: fc_in_channel = decoder_rnn_out_size + d_model + d_enc else: