From 6fbbacf94cdedc8a646fbc2c47ed836a31371d3c Mon Sep 17 00:00:00 2001
From: dohe0342 <dohe1070@gmail.com>
Date: Wed, 24 May 2023 20:59:02 +0900
Subject: [PATCH] from local

---
 .../.train_lora.py.swp                        | Bin 86016 -> 90112 bytes
 .../train_lora.py                             |   5 +++++
 2 files changed, 5 insertions(+)

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 81a9844df3505a8ef1719ec8cf7280540113b440..02385d22eda8801eebc9e89dbabb3e2fcb473cf4 100644
GIT binary patch
delta 501
zcmZ9|u}cDR6bJBk-dTjH9VSsrznT={u%@uwYDfuMYPCzxI`r=J)Tp4&EgG7nF^C|V
ziuC?~8loj6B_e2OZHj_gqM^Qr1l1Ql{J8gi?|1Mj2ky#+JGc~Cn3-kc(kLN4M5mux
z`l=Ru_H(^9-=lF{yjl?Z#Gung2HFR0+Uw%7ug-)sZ>K<%aHEdtxGgpZIqLzR=dcY<
z>jZW|fk_yG0Px^~2A(KD0PcCBBiMr&bis#<=oWTh5yrp=mmE<YYVa#|qqWahU!{PR
zGfY`3%BxB;wW6f5c|+ApA$5f1WmDFQlCqXhrd2ac>|Zj5$#BgXXR6K=y{svwTv9?#
zC8-(da!x6Rqkrbq3^Qdt%VIGWi869}_T$~~h9#$o3=h#8I7|B3;g_H21#aLNwjc%b
tFbz}S2iu1l^uv=E8v+XwFabfh?j|~f63jsdytn^pyM(gGZniuEN8fs+V<-Rs

delta 192
zcmZoTz}m2YRW!*U%+puFLeG!^2m}}y=Kao1nXysyEkC31W@dq0M#lKbg@WRnR|rP%
zO?;rRNr7bpuQ^cVc}9rx_|1YB<QZd6vobKO2I6KQjsjwPAT|VIMIcrH;@2z;4E;cC
z2*h8R85j-&@n#^724Zy}76;;wObiUmfw&Hc1Atfuh#vv1xiEQQqu#^^qT9bjGjg*`
N&bh&}`T14JDga}+D~137

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 3371a7b4e..1399f61fd 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
@@ -1142,6 +1142,11 @@ def train_one_epoch(
                 rank=rank,
             )
             del params.cur_batch_idx
+
+            if rank == 0: 
+                for i, lora in enumerate(lora_modules):
+                    lora.save_checkpoint(i, params.exp_dir)
+
             '''
             remove_checkpoints(
                 out_dir=params.exp_dir,