From f05b3b17018f55fe1652d16a8f4d86fa75452369 Mon Sep 17 00:00:00 2001 From: Yifan Yang <64255737+yfyeung@users.noreply.github.com> Date: Sat, 7 Sep 2024 23:13:41 +0800 Subject: [PATCH] Update finetune.py --- egs/librispeech/SSL/hubert/finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/SSL/hubert/finetune.py b/egs/librispeech/SSL/hubert/finetune.py index 05b942f63..582771dee 100755 --- 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: