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
6f6e91f992
commit
b8232df095
Binary file not shown.
@ -153,7 +153,7 @@ class ResidualAdapterModule(nn.Module):
|
||||
[build_adapter(embedding_dim, proj_dim) for _ in range(layer_num)]
|
||||
)
|
||||
|
||||
def forward(x, layer_id):
|
||||
def forward(x, layer_id=-1):
|
||||
x = x.transpose(0, 1)
|
||||
residual = x
|
||||
x = self.adapter_layers[layer_id](x)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user