提交 24633556 编写于 作者: W WenmuZhou

hidden_size添加默认参数

上级 f5613aa9
...@@ -116,7 +116,7 @@ class EncoderWithFC(nn.Layer): ...@@ -116,7 +116,7 @@ class EncoderWithFC(nn.Layer):
class SequenceEncoder(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__() super(SequenceEncoder, self).__init__()
self.encoder_reshape = EncoderWithReshape(in_channels) self.encoder_reshape = EncoderWithReshape(in_channels)
self.out_channels = self.encoder_reshape.out_channels self.out_channels = self.encoder_reshape.out_channels
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册