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