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
2302449e95
commit
a08df1ed6b
Binary file not shown.
@ -41,7 +41,6 @@ class TransformerEncoderAdapter(TransformerEncoder):
|
|||||||
self.adapters = ResidualAdapterModule()
|
self.adapters = ResidualAdapterModule()
|
||||||
for p in self.adapters.parameters():
|
for p in self.adapters.parameters():
|
||||||
p.data = nn.Parameter(torch.zeros(p.size()).to('cuda'))
|
p.data = nn.Parameter(torch.zeros(p.size()).to('cuda'))
|
||||||
print(p)
|
|
||||||
|
|
||||||
def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):
|
def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):
|
||||||
x, layer_results = self.extract_features_with_adapter(
|
x, layer_results = self.extract_features_with_adapter(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user