mirrors/icefall
Archived
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:03:24 +09:00
parent 6c60083ea9
commit 8ff991d95a
2 changed files with 0 additions and 9 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_encoder.py.swp
View File

Binary file not shown.

9
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
View File

@ -73,15 +73,6 @@ class FairSeqData2VecEncoder(EncoderInterface):
model.feature_grad_mult = 0.0 ## for conv network freeze model.feature_grad_mult = 0.0 ## for conv network freeze
model.mask_prob = 0.5 ## for conv network freeze model.mask_prob = 0.5 ## for conv network freeze
if not isinstance(model, Wav2Vec2Model):
try:
model = model.w2v_encoder.w2v_model
except:
print(
"using data2vec ..."
)
self.encoders = model self.encoders = model
self.pretrained_params = copy.deepcopy(model.state_dict()) self.pretrained_params = copy.deepcopy(model.state_dict())