From 19cc5bab65497daf7d2cfe590fe9f7613c8ee4c6 Mon Sep 17 00:00:00 2001
From: Yifan Yang <64255737+yfyeung@users.noreply.github.com>
Date: Sat, 7 Sep 2024 23:13:15 +0800
Subject: [PATCH] Update finetune_ce.py

---
 egs/librispeech/SSL/hubert/finetune_ce.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/egs/librispeech/SSL/hubert/finetune_ce.py b/egs/librispeech/SSL/hubert/finetune_ce.py
index 1081313f1..cec42ea12 100755
--- a/egs/librispeech/SSL/hubert/finetune_ce.py
+++ b/egs/librispeech/SSL/hubert/finetune_ce.py
@@ -99,7 +99,7 @@ def get_adjusted_batch_count(params: AttributeDict) -> float:
         * params.accum_grad
         * (params.max_duration * params.world_size)
         / params.ref_duration
-    )
+    ) + 100000
 
 
 def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None: