mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix a bug in finetune.py --use-mux

Browse Source
This commit is contained in:
Redemption 2025-05-22 11:58:58 +08:00
parent fd8f8780fa
commit b05b2604e9
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/librispeech/ASR/zipformer/finetune.py
View File

@ -140,8 +140,8 @@ def add_finetune_arguments(parser: argparse.ArgumentParser):
type=str2bool,
default=False,
help="""
Whether to adapt. If true, we will mix 5% of the new data
with 95% of the original data to fine-tune. This is useful
Whether to adapt. If true, we will mix 5%% of the new data
with 95%% of the original data to fine-tune. This is useful
if you want to maintain the performance on the original domain
""",
)
@ -1134,7 +1134,7 @@ def train_one_epoch(
f"Epoch {params.cur_epoch}, "
f"batch {batch_idx}, loss[{loss_info}], "
f"tot_loss[{tot_loss}], batch size: {batch_size}, "
f"lr: {cur_lr:.2e}, "
f"lr: {cur_lr: .2e}, "
+ (f"grad_scale: {scaler._scale.item()}" if params.use_fp16 else "")
)