skipping batch counts hurts performance

k2-fsa · Oct 29, 2024 · 1b89c6d · 1b89c6d
1 parent d4b6cb0
commit 1b89c6d
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/egs/librispeech/SSL/hubert/finetune.py b/egs/librispeech/SSL/hubert/finetune.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:

diff --git 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:

diff --git a/egs/librispeech/SSL/zipformer/finetune.py b/egs/librispeech/SSL/zipformer/finetune.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:

diff --git a/egs/librispeech/SSL/zipformer_ctc/finetune_ctc.py b/egs/librispeech/SSL/zipformer_ctc/finetune_ctc.py
@@ -93,7 +93,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: