mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
6c60083ea9
commit
8ff991d95a
Binary file not shown.
@ -73,15 +73,6 @@ class FairSeqData2VecEncoder(EncoderInterface):
|
||||
model.feature_grad_mult = 0.0 ## 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.pretrained_params = copy.deepcopy(model.state_dict())
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user