From 2463355632c9dbcc859a57a2a97dbfdfeaea0b95 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Fri, 16 Oct 2020 16:39:37 +0800 Subject: [PATCH] =?UTF-8?q?hidden=5Fsize=E6=B7=BB=E5=8A=A0=E9=BB=98?= =?UTF-8?q?=E8=AE=A4=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ppocr/modeling/necks/rnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/modeling/necks/rnn.py b/ppocr/modeling/necks/rnn.py index 8a744e0d..004ec564 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 -- GitLab