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)]
|
[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)
|
x = x.transpose(0, 1)
|
||||||
residual = x
|
residual = x
|
||||||
x = self.adapter_layers[layer_id](x)
|
x = self.adapter_layers[layer_id](x)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user