Skip to content

GitLab

Unverified Commit f87f8219 authored by andyjpaddle's avatar andyjpaddle Committed by GitHub
Browse files

fix sar bug (#5867)

parent 439c1224
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/modeling/heads/rec_sar_head.py
View file @ f87f8219
...@@ -216,7 +216,7 @@ class ParallelSARDecoder(BaseDecoder): ...@@ -216,7 +216,7 @@ class ParallelSARDecoder(BaseDecoder):
self.pred_dropout = nn.Dropout(pred_dropout) self.pred_dropout = nn.Dropout(pred_dropout)
pred_num_classes = self.num_classes - 1 pred_num_classes = self.num_classes - 1
if pred_concat: 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: else:
fc_in_channel = d_model fc_in_channel = d_model
self.prediction = nn.Linear(fc_in_channel, pred_num_classes) self.prediction = nn.Linear(fc_in_channel, pred_num_classes)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment