from local

This commit is contained in:
dohe0342 2023-03-15 12:19:09 +09:00
parent 989cbbe4b4
commit 1634efa32c
2 changed files with 1 additions and 1 deletions

View File

@ -280,7 +280,7 @@ class Data2VecAudioModel(BaseFairseqModel):
torch.FloatTensor(cfg.encoder_embed_dim).uniform_() torch.FloatTensor(cfg.encoder_embed_dim).uniform_()
) )
s#elf.encoder = TransformerEncoder(cfg) #elf.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)