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