Merge pull request #1890 from tink2123/srn_ch

Adaptation of Chinese char

Merge pull request #1890 from tink2123/srn_ch
Adaptation of Chinese char
0c5c9f69 · xiaoting · GitHub · 2a0c3d4d · 7f2304ab · 0c5c9f69
Unverified Commit 0c5c9f69 authored Feb 01, 2021 by xiaoting Committed by GitHub Feb 01, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ppocr/modeling/heads/rec_srn_head.py ppocr/modeling/heads/rec_srn_head.py +1 -1

No files found.
--- a/ppocr/modeling/heads/rec_srn_head.py
+++ b/ppocr/modeling/heads/rec_srn_head.py
@@ -246,7 +246,7 @@ class SRNHead(nn.Layer):
            num_encoder_tus=self.num_encoder_TUs,
            num_decoder_tus=self.num_decoder_TUs,
            hidden_dims=self.hidden_dims)
-        self.vsfd = VSFD(in_channels=in_channels)
+        self.vsfd = VSFD(in_channels=in_channels, char_num=self.char_num)

        self.gsrm.wrap_encoder1.prepare_decoder.emb0 = self.gsrm.wrap_encoder0.prepare_decoder.emb0