From d5f2ab367b0fa1ca431f4a2496eae1aced50aea6 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 21:22:50 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 90112 -> 90112 bytes .../train_lora.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.train_lora.py.swp index 534428f8766a005626f68c08886fcacbc3692c52..79e226fe58180ee9147e4a1e4746c6634010f7ad 100644 GIT binary patch delta 68 zcmZoTz}j$tRW!*U%+puFLeG!^2m}}yrZMKF9N8%P-kx;@4+DeBWKM^4)+}BiYjdf? WdY;LuzS)fN%@cgLPw-_d5CQ;u5fd-~ delta 68 zcmZoTz}j$tRW!*U%+puFLeG!^2m}}yIvDd(mTVM#Z_m1thk-$LGN(g2Yc?;CwYk(` WJr862WEJ1+<_W&rC-^cJ2mt_ce-i}& diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py index 9331721c5..7a5f95973 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/train_lora.py @@ -142,7 +142,7 @@ class LoRAHook(): lora_out = self.lora(input) output = input + lora_out - def save_checkpoint(self, i, iter, save_dir): + def save_checkpoint(self, i, iter_, save_dir): torch.save(self.lora.state_dict(), f"{save_dir}/lora_{i}.pt")