Commit 0140d73d by xuchen

fix the bug of conformer

parent 6c6d089a
...@@ -130,8 +130,8 @@ def base_architecture(args): ...@@ -130,8 +130,8 @@ def base_architecture(args):
args.conv_channels = getattr(args, "conv_channels", 1024) args.conv_channels = getattr(args, "conv_channels", 1024)
# Conformer # Conformer
args.macaron_style = getattr(args, "macaron_style", True) args.macaron_style = getattr(args, "macaron_style", True)
args.macaron_style = getattr(args, "use_cnn_module", True) args.use_cnn_module = getattr(args, "use_cnn_module", True)
args.macaron_style = getattr(args, "cnn_module_kernel", 31) args.cnn_module_kernel = getattr(args, "cnn_module_kernel", 31)
args.encoder_embed_dim = getattr(args, "encoder_embed_dim", 512) args.encoder_embed_dim = getattr(args, "encoder_embed_dim", 512)
args.encoder_ffn_embed_dim = getattr(args, "encoder_ffn_embed_dim", 2048) args.encoder_ffn_embed_dim = getattr(args, "encoder_ffn_embed_dim", 2048)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论