Skip to content

P
PaddleOCR

weixin_41840029 / PaddleOCR
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleOCR

通知 1
Star 1
Fork 0
P
PaddleOCR
提交 64ec4fb7 编写于 作者: A andyjpaddle
浏览文件
操作

fix sar bug

上级 25e2f818
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
ppocr/modeling/heads/rec_sar_head.py
浏览文件 @ 64ec4fb7
...@@ -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% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或