fix code style

2021-09-07 07:39:24 +00:00 · 2021-09-07 07:39:24 +00:00 · b68c895827
parent d611515803
commit b68c895827
1 changed files with 1 additions and 1 deletions
--- 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: