fix sar bug (#5864)

e758d273 · andyjpaddle · GitHub · e6133036 · e758d273
隐藏空白更改
内联并排

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
@@ -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)