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

Some temp code for loading old state dict

Browse Source
This commit is contained in:
Daniel Povey 2022-05-28 00:29:47 +08:00
parent d89cb53a3b
commit 828defb019
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/ASR/pruned_transducer_stateless4/train.py
View File

@ -875,13 +875,16 @@ def run(rank, world_size, args):
optimizer = Cain(model.parameters(), lr=params.initial_lr, optimizer = Cain(model.parameters(), lr=params.initial_lr,
max_eff_lr=params.initial_lr) max_eff_lr=params.initial_lr)
scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs) scheduler = Eden(optimizer, params.lr_batches, params.lr_epochs)
if checkpoints and "optimizer" in checkpoints: if checkpoints and "optimizer" in checkpoints:
logging.info("Loading optimizer state dict") logging.info("Loading optimizer state dict")
optimizer.load_state_dict(checkpoints["optimizer"]) optimizer.load_state_dict(checkpoints["optimizer"])
for p in optimizer.param_groups:
p['max_eff_lr'] = params.initial_lr # TEMP!!
if ( if (
checkpoints checkpoints
and "scheduler" in checkpoints and "scheduler" in checkpoints