From b2c0a0af16dcc34753724d3b27615f5fcc15556c Mon Sep 17 00:00:00 2001
From: dohe0342 <dohe1070@gmail.com>
Date: Wed, 24 May 2023 21:22:59 +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 79e226fe58180ee9147e4a1e4746c6634010f7ad..7fce0f924a59b7d7764ec94e74c03f48a9e2939a 100644
GIT binary patch
delta 68
zcmZoTz}j$tRW!*U%+puFLeG!^2m}}yW-;caoZBe+-kx;<4+DeDWKM@{Mw!i}4$FBa
XtN7ZpmZTQN*ET2kZcp%KTp$Dhj-3=1

delta 66
zcmZoTz}j$tRW!*U%+puFLeG!^2m}}yrZMKF9N8%P-kx;@4+DeBWKM@{MwQK_4$FBa
VuZ!oHEaPk6ob0<j*_Uyy5CClx6PEx0

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 7a5f95973..b9f59e2c5 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
@@ -143,7 +143,7 @@ class LoRAHook():
         output = input + lora_out
 
     def save_checkpoint(self, i, iter_, save_dir):
-        torch.save(self.lora.state_dict(), f"{save_dir}/lora_{i}.pt")
+        torch.save(self.lora.state_dict(), f"{save_dir}/lora_{iter_}_{i}.pt")
 
 
 def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None: