change CodeStyle, test=asr

paddlespeech/s2t/models/u2/u2.py

@@ -42,7 +42,8 @@ from paddlespeech.s2t.modules.cmvn import GlobalCMVN
 from paddlespeech.s2t.modules.ctc import CTCDecoderBase
 from paddlespeech.s2t.modules.decoder import BiTransformerDecoder
 from paddlespeech.s2t.modules.decoder import TransformerDecoder
-from paddlespeech.s2t.modules.encoder import ConformerEncoder, SqueezeformerEncoder
+from paddlespeech.s2t.modules.encoder import ConformerEncoder
+from paddlespeech.s2t.modules.encoder import SqueezeformerEncoder
 from paddlespeech.s2t.modules.encoder import TransformerEncoder
 from paddlespeech.s2t.modules.initializer import DefaultInitializerContext
 from paddlespeech.s2t.modules.loss import LabelSmoothingLoss

paddlespeech/s2t/modules/attention.py

@@ -26,7 +26,7 @@ from paddlespeech.s2t.utils.log import Log
 
 logger = Log(__name__).getlog()
 
-__all__ = ["MultiHeadedAttention", "RelPositionMultiHeadedAttention"]
+__all__ = ["MultiHeadedAttention", "RelPositionMultiHeadedAttention", "RelPositionMultiHeadedAttention2"]
 
 # Relative Positional Encodings
 # https://www.jianshu.com/p/c0608efcc26f

paddlespeech/s2t/modules/convolution.py

@@ -5,7 +5,7 @@ from paddle import nn
 from paddle.nn import initializer as I
 from typeguard import check_argument_types
 
-__all__ = ['ConvolutionModule']
+__all__ = ['ConvolutionModule2']
 
 from paddlespeech.s2t import masked_fill
 from paddlespeech.s2t.modules.align import Conv1D, BatchNorm1D, LayerNorm

paddlespeech/s2t/modules/encoder.py

@@ -43,7 +43,7 @@ from paddlespeech.s2t.utils.log import Log
 
 logger = Log(__name__).getlog()
 
-__all__ = ["BaseEncoder", 'TransformerEncoder', "ConformerEncoder"]
+__all__ = ["BaseEncoder", 'TransformerEncoder', "ConformerEncoder", "SqueezeformerEncoder"]
 
 
 class BaseEncoder(nn.Layer):

paddlespeech/s2t/modules/encoder_layer.py

@@ -26,7 +26,7 @@ from paddlespeech.s2t.utils.log import Log
 
 logger = Log(__name__).getlog()
 
-__all__ = ["TransformerEncoderLayer", "ConformerEncoderLayer"]
+__all__ = ["TransformerEncoderLayer", "ConformerEncoderLayer", "SqueezeformerEncoderLayer"]
 
 
 class TransformerEncoderLayer(nn.Layer):

paddlespeech/s2t/modules/positionwise_feed_forward.py

@@ -23,7 +23,7 @@ from paddlespeech.s2t.utils.log import Log
 
 logger = Log(__name__).getlog()
 
-__all__ = ["PositionwiseFeedForward"]
+__all__ = ["PositionwiseFeedForward", "PositionwiseFeedForward2"]
 
 
 class PositionwiseFeedForward(nn.Layer):

paddlespeech/s2t/modules/subsampling.py

@@ -32,7 +32,8 @@ logger = Log(__name__).getlog()
 
 __all__ = [
     "LinearNoSubsampling", "Conv2dSubsampling4", "Conv2dSubsampling6",
-    "Conv2dSubsampling8"
+    "Conv2dSubsampling8", "TimeReductionLayerStream", "TimeReductionLayer1D",
+    "TimeReductionLayer2D", "DepthwiseConv2DSubsampling4"
 ]