Update pretrain.py

2025-08-09 18:12:19 +00:00 · 2024-08-22 10:55:23 +08:00 · 2024-08-22 10:55:23 +08:00 · 6dbcdbac5d
commit 6dbcdbac5d
parent 8fe6713d10
1 changed files with 2 additions and 2 deletions
--- a/egs/librilight/SSL/zipformer/pretrain.py
+++ b/egs/librilight/SSL/zipformer/pretrain.py
@ -474,7 +474,7 @@ def get_parser():
    parser.add_argument(
        "--lr-hours",
        type=float,
-        default=30000,
+        default=10000,
        help="""Number of hours that affects how rapidly the learning rate decreases.
        """,
    )
@ -482,7 +482,7 @@ def get_parser():
    parser.add_argument(
        "--warmup-batches",
        type=float,
-        default=10000,
+        default=5000,
        help="Eden warmup steps",
    )