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
8fe13ee05b
commit
cdaa455848
Binary file not shown.
Binary file not shown.
@ -274,7 +274,8 @@ class Data2VecAudioModel(BaseFairseqModel):
|
||||
torch.FloatTensor(cfg.encoder_embed_dim).uniform_()
|
||||
)
|
||||
|
||||
self.encoder = TransformerEncoder(cfg)
|
||||
#self.encoder = TransformerEncoder(cfg)
|
||||
self.encoder = TransformerEncoderAdapter(cfg)
|
||||
self.layer_norm = LayerNorm(self.extractor_embed)
|
||||
|
||||
self.final_proj = nn.Linear(self.embed, self.embed)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user