from local

This commit is contained in:
dohe0342 2023-04-14 13:38:51 +09:00
parent e7328d2512
commit d0e589401f
3 changed files with 2 additions and 2 deletions

Binary file not shown.

View File

@ -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)