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

Fix bug: When resuming training from a checkpoint, model_avg was not assigned, resulting in a None error. (#1914)

Browse Source
This commit is contained in:
math345 2025-04-10 11:37:28 +08:00 committed by GitHub
parent 300a821f58
commit 64c5364085
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/wenetspeech/KWS/zipformer/finetune.py
View File

@ -593,6 +593,9 @@ def run(rank, world_size, args):
if params.continue_finetune: if params.continue_finetune:
assert params.start_epoch > 0, params.start_epoch assert params.start_epoch > 0, params.start_epoch
if rank == 0:
# model_avg is only used with rank 0
model_avg = copy.deepcopy(model).to(torch.float64)
checkpoints = load_checkpoint_if_available( checkpoints = load_checkpoint_if_available(
params=params, model=model, model_avg=model_avg params=params, model=model, model_avg=model_avg
) )