From 6dbcdbac5d57a32ca392139a917d076f6bba0aff Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Thu, 22 Aug 2024 10:55:23 +0800
Subject: [PATCH] Update pretrain.py

---
 egs/librilight/SSL/zipformer/pretrain.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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",
     )