Add back 2 conformer layers in 1st stack.

2022-12-20 13:54:06 +08:00 · 2022-12-20 13:54:06 +08:00 · d2b272ab50
commit d2b272ab50
parent 5fa8de5c05
1 changed files with 1 additions and 1 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py
@ -116,7 +116,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
    parser.add_argument(
        "--num-encoder-layers",
        type=str,
-        default="2,4,4,4,4,4",
+        default="4,4,4,4,4,4",
        help="Number of zipformer encoder layers per stack, comma separated.",
    )