fix code style

b68c8958 · andyjpaddle · d6115158 · b68c8958
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppocr/modeling/heads/rec_sar_head.py ppocr/modeling/heads/rec_sar_head.py +1 -1

未找到文件。
--- 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: