Skip to content

GitLab

"git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "ac98415b48218cfcd3fb609bbf091c73e3be1855"
Commit 8123688a authored by andyjpaddle's avatar andyjpaddle
Browse files

update rec_sar_head

parent f57a6d1b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ppocr/modeling/heads/rec_sar_head.py
View file @ 8123688a
...@@ -151,11 +151,11 @@ class ParallelSARDecoder(BaseDecoder): ...@@ -151,11 +151,11 @@ class ParallelSARDecoder(BaseDecoder):
**kwargs): **kwargs):
super().__init__() super().__init__()
self.num_classes = num_classes self.num_classes = out_channels
self.enc_bi_rnn = enc_bi_rnn self.enc_bi_rnn = enc_bi_rnn
self.d_k = d_k self.d_k = d_k
self.start_idx = out_channels - 2 self.start_idx = out_channels - 2
self.padding_idx = out_channels - 1 self.padding_idx = out_channels - 1
self.max_seq_len = max_text_length self.max_seq_len = max_text_length
self.mask = mask self.mask = mask
self.pred_concat = pred_concat self.pred_concat = pred_concat
......
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