diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp deleted file mode 100644 index 0edb6a499..000000000 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp and /dev/null differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py index 1470360fe..3732bcc2a 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py @@ -66,10 +66,10 @@ class FairSeqData2VecEncoder(EncoderInterface): model = models[0] model.feature_grad_mult = 0.0 ## for conv network freeze ## prevent overfitting - #model.mask_prob = 0.65 - #model.mask_channel_prob = 0.5 - #model.mask_channel_length = 64 - #model.activation_dropout = 0.1 + model.mask_prob = 0.65 + model.mask_channel_prob = 0.5 + model.mask_channel_length = 64 + model.activation_dropout = 0.1 self.encoders = model self.pretrained_params = copy.deepcopy(model.state_dict())