From 7be2b54d722b409db2affeae87c2d150f16b416b Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 21:19:54 +0900 Subject: [PATCH] from local --- .../.decode.py.swp | Bin 53248 -> 53248 bytes .../decode.py | 3 ++- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.decode.py.swp index c1e809bf2f784299b2f4c0a4cda087850493bf25..d6e6ce19f7fb264304f6ef5648abcf7fb0451938 100644 GIT binary patch delta 170 zcmZozz}&EaSv1KY%+puFLeG!^2m}}yrZVKE#B3D3Ex@R@`J;faKJQ;n28NAX5J|Pk zf)3N!9s_wBxi+tJ;NqIxpwH@=lV6mm2PG0y;)_cXOH$)gGLuU*K!WiFiA9OInhXj^ n03ohmt5A|(l$-(7pO`Y)(NlyMELWXbt5;B>q`6t$`(_IO?ASAc delta 81 zcmZozz}&EaSv1KY%+puFLeG!^2m}}y`Wf<40yc`?7GPA}{82zypEsL}fkB!ZBB?rA d&|x|oh$qUud7T3n*JKCRTqXsD&FbDaTLAOE62Slf diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py index 980083656..597a46995 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py @@ -697,7 +697,8 @@ def main(): lora_modules.append(LoRAHook(module)) for i, lora in enumerate(lora_modules): - + lora_param = torch.load(f"{params.exp_dir}/lora_{i}.pt") + lora.lora.load_state_dict(lora_param) else: if not params.use_averaged_model: if params.iter > 0: