hidden_size添加默认参数

24633556 · WenmuZhou · f5613aa9 · 24633556
Commit 24633556 authored Oct 16, 2020 by WenmuZhou
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ppocr/modeling/necks/rnn.py ppocr/modeling/necks/rnn.py +1 -1

No files found.
--- 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