Reduce feedforward-dims

2025-12-11 06:55:27 +00:00 · 2023-01-06 22:26:58 +08:00 · 2023-01-06 22:26:58 +08:00 · ef48019d6e
commit ef48019d6e
parent 9b0c0aabb2
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
@ -138,7 +138,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
    parser.add_argument(
        "--feedforward-dim",
        type=str,
-        default="1792,1792,2304,2304,2304,1792",
+        default="1536,1536,1536,1536,1536,1536",
        help="Feedforward dimension of the zipformer encoder layers, per stack, comma separated.",
    )