From dfe4a02a25d68c9ee05bc3837a212cc9ada630c0 Mon Sep 17 00:00:00 2001
From: dohe0342 <dohe1070@gmail.com>
Date: Wed, 24 May 2023 21:22:47 +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 870d27e4258070d0ef64c539d53a8309fdbf1e43..534428f8766a005626f68c08886fcacbc3692c52 100644
GIT binary patch
delta 72
zcmZoTz}j$tRV>LM%+puFLeG!^2m}}yIvDd(mPl<BduPwOl81pobuy<zI%_sBkhQtg
aVFS-(ZQpd3%#zfi<_W&rC-^cJ2mt`SToil&

delta 73
zcmZoTz}j$tRV>LM%+puFLeG!^2m}}yLjL8Ze3aZM_RgMlJr4te_GC_nbk;&%AZv4}
b!v-GKq{NcsjL9m#8O@V?w@>nA%o73t2R{|_

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 52ab06ee0..9331721c5 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, save_dir):
+    def save_checkpoint(self, i, iter, save_dir):
         torch.save(self.lora.state_dict(), f"{save_dir}/lora_{i}.pt")