mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
8435ec33f7
commit
28bc046e2a
Binary file not shown.
@ -163,7 +163,7 @@ class LoRAModule(nn.Module):
|
||||
|
||||
def reset_parameters(self):
|
||||
nn.init.zeros_(self.lora_B.weight.data)
|
||||
nn.init_normal_(self.lora_A.weight.dat)
|
||||
nn.init.normal_(self.lora_A.weight.data)
|
||||
|
||||
def forward(self, x):
|
||||
x = x.transpose(0, 1)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user