From 8435ec33f7e37a3c903819927534d871cc4f3bbb Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Wed, 24 May 2023 13:33:36 +0900 Subject: [PATCH] from local --- .../.data2vec_audio.py.swp | Bin 45056 -> 45056 bytes .../.train_lora.py.swp | Bin 90112 -> 90112 bytes .../data2vec_audio.py | 4 ++-- 3 files changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp index e191309b5674aa0af0f7168af0da6471d0efb796..80dae400201297d0628e52fa82b22e9d609ae1a0 100644 GIT binary patch delta 96 zcmZp8z|`=7X#mLpVh9{Fb9kwg(=3-#z2I4LtE&$>nAT|eL86f`02~@TDp#wMP iWM;2J+?h}@$TW&>6=i%K4*;nh2~Pk3 diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py index b6997faac..d089daf10 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py @@ -162,8 +162,8 @@ class LoRAModule(nn.Module): self.reset_parameters() def reset_parameters(self): - nn.init.zeros_(self.lora_B) - nn.init_normal_(self.lora_A) + nn.init.zeros_(self.lora_B.weight.data) + nn.init_normal_(self.lora_A.weight.dat) def forward(self, x): x = x.transpose(0, 1)