From 9b1c9001b29c05cf8b9bd09b77415cb8e6851936 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 25 May 2023 17:23:59 +0900 Subject: [PATCH] from local --- .../.train_lora.py.swp | Bin 86016 -> 86016 bytes .../train_lora.py | 2 ++ 2 files changed, 2 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 fd598a85bee7c6bdd4875c4dad22458c3cc6c25b..1a93cba027ef1eba6a7bf02975e33b205f9c0d0a 100644 GIT binary patch delta 589 zcmXxhO(??w7{~Ev__K}iM_x)HF4PXzri8-DYk6BC2|IWxY_!FNzful%T03mXvQo&2 z=Dn1hgiwld!qr>3h?RUDJoWkY9G=VbJhNeCHmq!xs5)Q7qTV!{szoFmL#QG?e`;U1 zc3~1ZM-UCDM;*SBMDDSNMNFazd3Z|{dBqDh5x@)v(SlrLBOTvHktbYY7e08gfN`T} z(nHe-D>9)X1+m zA$G8YIXKY)9is7Ufk&KT8zI!929+p;0iTM<16B}15E=$xM-|dxf*$W$oX8pWv5GL3 zF^?g%q5`F`AqPLPoE~@B!xq-DhEdpILje>V^YB9);P3ex_^&|JqBHqD4s+zhVNJ|( cO$?8@i$8;f0-FSxoBG{2Su7KN=jMb;^JaTqt*^G>|m_S!FI#N@H^nb4$8p} zG6zc*rI1pHVMTM30}h&U@pbdo=jl1TwI9OMfS0QIgFznt!Rb~I=q*NWU-4Rwy}&EjFm`K zdMPb%!HGr`Od?OXz#i5yjY)XXjV9PpC>Hs|2eQZ@i8Vwpfe`xPMja}VFA~WWMMdr? zhuFj-rceVrY|!G>C~}JohT+E`I#G`*XmDcS2WccQi!gj>h68m_@TM1eh!Uqb#u2tK zj}QjX2M^j%4GX??%!6y3;|vQ3!-EdgBG1q{JmPQk2*&aMmYz*3J6#&R8umA+zG0(! T>9?x+AxG-cZ#1XAg65ZBSr%J9 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 a60e2dcbb..4ec530106 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 @@ -141,6 +141,8 @@ class LoRAHook(): lora_out = self.lora(input) output += lora_out + print(lora_out, output) + def save_checkpoint(self, i, iter_, save_dir): torch.save(self.lora.state_dict(), f"{save_dir}/lora_{iter_}_{i}.pt")