hidden_size添加默认参数

2020-10-16 16:39:37 +08:00 · 2020-10-16 16:39:37 +08:00 · 2463355632
parent f5613aa9be
commit 2463355632
1 changed files with 1 additions and 1 deletions
--- a/ppocr/modeling/necks/rnn.py
+++ b/ppocr/modeling/necks/rnn.py
@ -116,7 +116,7 @@ class EncoderWithFC(nn.Layer):


 class SequenceEncoder(nn.Layer):
-    def __init__(self, in_channels, encoder_type, hidden_size, **kwargs):
+    def __init__(self, in_channels, encoder_type, hidden_size=48, **kwargs):
        super(SequenceEncoder, self).__init__()
        self.encoder_reshape = EncoderWithReshape(in_channels)
        self.out_channels = self.encoder_reshape.out_channels