from local

egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py

@@ -125,6 +125,13 @@ def add_model_arguments(parser: argparse.ArgumentParser):
         help="Number of attention heads in the zipformer encoder layers.",
     )
     
+    parser.add_argument(
+        "--encoder-dim",
+        type=int,
+        default=768,
+        help="Encoder embedding dimension",
+    )
+
     parser.add_argument(
         "--encoder-dims",
         type=str,