from local

This commit is contained in:
dohe0342 2022-12-27 11:12:22 +09:00
parent 38c47c14e8
commit 7f2f7806c0
3 changed files with 3 additions and 1 deletions

View File

@ -278,7 +278,8 @@ class Data2VecAudioModel(BaseFairseqModel):
self.final_proj = nn.Linear(self.embed, self.embed)
self.num_updates = 0
'''
def make_ema_teacher(self):
ema_config = EMAModuleConfig(
ema_decay=self.cfg.ema_decay,
@ -295,6 +296,7 @@ class Data2VecAudioModel(BaseFairseqModel):
ema_config,
skip_keys=skip_keys,
)
'''
def set_num_updates(self, num_updates):
super().set_num_updates(num_updates)

Binary file not shown.