From 0c78b6761ba1a7c6f4f3466b705bb16aa9e2d57d Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 13 Dec 2022 11:40:25 +0900 Subject: [PATCH] from local --- .../.train.py.swp | Bin 69632 -> 69632 bytes .../train.py | 6 +++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train.py.swp index dfdd8823962172a56f4b2cea0e87eb7c5160ab04..b9df3c8b73e8b879159ac13d810ea42a07066cfe 100644 GIT binary patch delta 170 zcmZozz|ydQMJ&l6%+puFQqO<^2m}}yUOt_k?7nBCSd<{6|K>u$$$GrXEDQ|QtPolM z&4Ld4jDkFD3=D68LVJO@4T!5Z2Rdr7FexZZ-gQZOvW^?4ZnRQyaz<)OX-;ZUd}>~@ yQmjH|u|i&ciGp8#UMhnE5=hKTnOyBAKRI!!@Z=Y+yu5J9%rpg!%@S@eP67aDzAYpG delta 122 zcmZozz|ydQMJ&l6%+puFQqO<^2m}}yRGv&vHrTmQEJ~2kdvl@SWIf(~76yjvtPol6 z&4Ld4jDjHU79jQqVn-n6*c|An!7}-Q8}H=#@p_ZbJ&~O(f0dO{3C5ay)>UqLKQ|-W JW?8q#CjoJPBM1Nh diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py index 508fca068..df7083260 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train.py @@ -1311,7 +1311,11 @@ def run(rank, world_size, args, wb=None): and ("scheduler" in checkpoints or "scheduler_enc" in checkpoints) and checkpoints["scheduler"] is not None ): - if params.multi_optim and "scheduler_enc" in checkpoints: + if ( + params.multi_optim + and "scheduler_enc" in checkpoints + and checkpoin["scheduler_enc"] is not None + ): scheduler_enc.load_state_dict(checkpoints["scheduler_enc"]) scheduler_dec.load_state_dict(checkpoints["scheduler_dec"]) else: