From debf2ded038493f9b7a9f992dbac7e31c5a1b875 Mon Sep 17 00:00:00 2001
From: dohe0342 <dohe1070@gmail.com>
Date: Thu, 25 May 2023 21:31:47 +0900
Subject: [PATCH] from local

---
 .../.train_lora.py.swp                        | Bin 86016 -> 86016 bytes
 .../train_lora.py                             |   1 +
 2 files changed, 1 insertion(+)

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 e986546b0410c876bc796624f117c249b9af65a8..854a0a764d49ac219892c2041e660d6f91c84bf4 100644
GIT binary patch
delta 467
zcmXBQPbkB27{~GFyV)Q6nfCi-2Z~mRl$n%vTFOdOZsLM6qp7vuLgTg-$wAUCJGksJ
ziPIvxIdG6D<g}~OE;|j0UA*I|*V9wabNKXR3$|>*=8XnsB2&Jge^@AwNPXIj<xAd%
zuA<wzQPvIh-sx5rk)rkgAez?$sylPvDzc6wW-*2~yt+khaD+{ykwh38sx9nMKn_W)
zU;;ka@uQIgx41+cVfgUo61jFsMlKi*v4<EU=z{~5W|2pvU}6XZ_;!kXpoR(#u!A5v
zaOV)&LJB7QsM`sF5_U0<IeS8+i=iD28;S9N9QLt*NsJ?eQS`vTR}){t6RvQMGn^uh
tUetJY6~`!AOV3VA`DH{6YWb6p@)!woZPEYNHq8Fma(Z>C{25Z3{tL$iP$&QZ

delta 437
zcmXBQPbk9y7{>AEx7o0n`K|4+gGOX7H0Cg;<wDX9v^a2?KcysA;<yzT<RC4FZ8c%{
zB`ry=E)M3vNwlMb6)B;7%Tu3v4)5iC(gj<(VDrs}CP!m!k={O$7U?LyGQU^yeVayN
z)AW>0kMZ8>>bFQ0>;FOG&<q(JsY0`q#TJ$^1UG6f=>{h_Mixm#;6<%Ty2UlJIKUcW
z2qTC9KAch&86**d2X7AP#i5k0DCgM6I^r0B9Y2jy9y!cl3SDT2A9h?eNF^*_1fTWN
zDe}l+8Ua-8(i6_GgKdo27o}lJH-h-G@ewYtiw&$|33C`n6rFJ6w@!M+Gam7P3?|@*
W7xxTav6k)zE#+d^=y4XKhWQ85vqZH3

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 1d31becfa..0db6602fe 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
@@ -1615,6 +1615,7 @@ def run_adapter(rank, world_size, args, wb=None):
     if world_size > 1:
         logging.info("Using DDP for LoRA")
         for lora in lora_modules:
+            lora.lora = lora.lora.to(device)
             lora.lora = DDP(lora.lora, device_ids=[rank], find_unused_parameters=False)
     
     adapter_names = []