Skip to content

GitLab

Commit b68c8958 authored by andyjpaddle's avatar andyjpaddle
Browse files

fix code style

parent d6115158
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/modeling/heads/rec_sar_head.py
View file @ b68c8958
...@@ -195,7 +195,7 @@ class ParallelSARDecoder(BaseDecoder): ...@@ -195,7 +195,7 @@ class ParallelSARDecoder(BaseDecoder):
# Prediction layer # Prediction layer
self.pred_dropout = nn.Dropout(pred_dropout) self.pred_dropout = nn.Dropout(pred_dropout)
pred_num_classes = 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 + d_enc
else: else:
......
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