mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-20 16:04:40 +09:00
parent 4e85e32ffd
commit 46e95a415d
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp Normal file
View File

Binary file not shown.

5
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
View File

@ -165,7 +165,8 @@ class Data2VecAudioModel(BaseFairseqModel):
def set_num_updates(self, num_updates): def set_num_updates(self, num_updates):
super().set_num_updates(num_updates) super().set_num_updates(num_updates)
'''
if self.ema is None and self.final_proj is not None: if self.ema is None and self.final_proj is not None:
logger.info(f"making ema teacher") logger.info(f"making ema teacher")
self.make_ema_teacher() self.make_ema_teacher()
@ -183,7 +184,7 @@ class Data2VecAudioModel(BaseFairseqModel):
self.ema.set_decay(decay) self.ema.set_decay(decay)
if self.ema.get_decay() < 1: if self.ema.get_decay() < 1:
self.ema.step(self.encoder if self.cfg.ema_transformer_only else self) self.ema.step(self.encoder if self.cfg.ema_transformer_only else self)
'''
self.num_updates = num_updates self.num_updates = num_updates
def state_dict(self, destination=None, prefix="", keep_vars=False): def state_dict(self, destination=None, prefix="", keep_vars=False):