fix sar bug (#5864)

2022-04-01 19:34:44 +08:00 · 2022-04-01 19:34:44 +08:00 · e758d27318
parent e6133036b4
commit e758d27318
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
@ -216,7 +216,7 @@ class ParallelSARDecoder(BaseDecoder):
        self.pred_dropout = nn.Dropout(pred_dropout)
        pred_num_classes = self.num_classes - 1
        if pred_concat:
-            fc_in_channel = decoder_rnn_out_size + d_model + d_enc
+            fc_in_channel = decoder_rnn_out_size + d_model + encoder_rnn_out_size
        else:
            fc_in_channel = d_model
        self.prediction = nn.Linear(fc_in_channel, pred_num_classes)