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
b21111b67f
commit
0207b96426
Binary file not shown.
@ -145,9 +145,11 @@ class ResidualAdapterModule(nn.Module):
|
||||
|
||||
super().__init__()
|
||||
|
||||
self.adapters = ConvolutionModule(768, 31)
|
||||
def build_adapter(embedding_dim, proj_dim, type_='conv'):
|
||||
return nn.Sequential(
|
||||
if type_ == 'conv':
|
||||
return ConvolutionModule(768, 31)
|
||||
else:
|
||||
return nn.Sequential(
|
||||
#nn.LayerNorm(embedding_dim),
|
||||
nn.Linear(embedding_dim, proj_dim),
|
||||
nn.ReLU(),
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user