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
7d07640d39
commit
38e30d98b1
Binary file not shown.
@ -40,7 +40,7 @@ class TransformerEncoderAdapter(TransformerEncoder):
|
||||
super().__init__(args)
|
||||
self.adapters = ResidualAdapterModule()
|
||||
for p in self.adapters.parameters():
|
||||
p.data = torch.nn.parameter(torch.zeros(p.size()).to('cuda'))
|
||||
p.data = torch.nn.parameter.Parameter(torch.zeros(p.size()).to('cuda'))
|
||||
print(p)
|
||||
|
||||
def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user