From 8247eba840c0b0a154e9eee67400ad48e4d87697 Mon Sep 17 00:00:00 2001 From: zxcd <228587199@qq.com> Date: Wed, 3 Sep 2025 19:36:40 +0800 Subject: [PATCH] add paramters name (#4119) --- paddlespeech/s2t/modules/align.py | 53 ++++++++++++++++++++++++------- 1 file changed, 42 insertions(+), 11 deletions(-) diff --git a/paddlespeech/s2t/modules/align.py b/paddlespeech/s2t/modules/align.py index a206a350c..30776426f 100644 --- a/paddlespeech/s2t/modules/align.py +++ b/paddlespeech/s2t/modules/align.py @@ -35,8 +35,12 @@ class LayerNorm(nn.LayerNorm): if bias_attr is None: bias_attr = paddle.ParamAttr( initializer=nn.initializer.Constant(0.0)) - super(LayerNorm, self).__init__(normalized_shape, epsilon, weight_attr, - bias_attr, name) + super(LayerNorm, self).__init__( + normalized_shape=normalized_shape, + epsilon=epsilon, + weight_attr=weight_attr, + bias_attr=bias_attr, + name=name) class BatchNorm1D(nn.BatchNorm1D): @@ -54,9 +58,14 @@ class BatchNorm1D(nn.BatchNorm1D): if bias_attr is None: bias_attr = paddle.ParamAttr( initializer=nn.initializer.Constant(0.0)) - super(BatchNorm1D, - self).__init__(num_features, momentum, epsilon, weight_attr, - bias_attr, data_format, name) + super(BatchNorm1D, self).__init__( + num_features=num_features, + momentum=momentum, + epsilon=epsilon, + weight_attr=weight_attr, + bias_attr=bias_attr, + data_format=data_format, + name=name) class Embedding(nn.Embedding): @@ -99,8 +108,12 @@ class Linear(nn.Linear): fan_in=None, negative_slope=math.sqrt(5), nonlinearity='leaky_relu')) - super(Linear, self).__init__(in_features, out_features, weight_attr, - bias_attr, name) + super(Linear, self).__init__( + in_features=in_features, + out_features=out_features, + weight_attr=weight_attr, + bias_attr=bias_attr, + name=name) class Conv1D(nn.Conv1D): @@ -131,8 +144,17 @@ class Conv1D(nn.Conv1D): negative_slope=math.sqrt(5), nonlinearity='leaky_relu')) super(Conv1D, self).__init__( - in_channels, out_channels, kernel_size, stride, padding, dilation, - groups, padding_mode, weight_attr, bias_attr, data_format) + in_channels=in_channels, + out_channels=out_channels, + kernel_size=kernel_size, + stride=stride, + padding=padding, + dilation=dilation, + groups=groups, + padding_mode=padding_mode, + weight_attr=weight_attr, + bias_attr=bias_attr, + data_format=data_format) class Conv2D(nn.Conv2D): @@ -163,5 +185,14 @@ class Conv2D(nn.Conv2D): negative_slope=math.sqrt(5), nonlinearity='leaky_relu')) super(Conv2D, self).__init__( - in_channels, out_channels, kernel_size, stride, padding, dilation, - groups, padding_mode, weight_attr, bias_attr, data_format) + in_channels=in_channels, + out_channels=out_channels, + kernel_size=kernel_size, + stride=stride, + padding=padding, + dilation=dilation, + groups=groups, + padding_mode=padding_mode, + weight_attr=weight_attr, + bias_attr=bias_attr, + data_format=data_format)