Update pretrain.py

2025-08-09 18:12:19 +00:00 · 2024-08-22 18:32:33 +08:00 · 2024-08-22 18:32:33 +08:00 · ad61d72dfa
commit ad61d72dfa
parent 6357d420b7
1 changed files with 1 additions and 1 deletions
--- a/egs/librilight/SSL/zipformer/pretrain.py
+++ b/egs/librilight/SSL/zipformer/pretrain.py
@ -482,7 +482,7 @@ def get_parser():
    parser.add_argument(
        "--warmup-batches",
        type=float,
-        default=5000,
+        default=1000,
        help="Eden warmup steps",
    )