diff --git a/egs/librilight/SSL/zipformer/pretrain.py b/egs/librilight/SSL/zipformer/pretrain.py index 93613be5a..f5c98f7eb 100755 --- 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", )