diff --git a/ppocr/modeling/necks/rnn.py b/ppocr/modeling/necks/rnn.py index 8a744e0d67ad902cfb52cbf1a0eb02e897c61d0b..004ec5641c2c4c33037e5a9af38ad255331a67f5 100644 --- 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