From 8ff991d95a28ad422334ea25cd47db5e5300e57a Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 20 Dec 2022 16:03:24 +0900 Subject: [PATCH] from local --- .../.data2vec_encoder.py.swp | Bin 16384 -> 16384 bytes .../data2vec_encoder.py | 9 --------- 2 files changed, 9 deletions(-) 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 index 30a6bc65850b48cce9c1cec8f386696244b31759..5db563cbf84b6f463bb0be6e5f6c8cf922c06884 100644 GIT binary patch delta 303 zcmZo@U~Fh$6iYG)^Ym4))H7fJ0s#hw{@8`d@);Y&e(^IVZx$5z%+IUJ#K3Tt86pc5 zQZQz$U&O}15D&z0Kr9Tz?^zibb^&od5ZeKfH($- zt%2AIh&6#&8i<*I_zBP&kX2`Zcoz^)1mY?nP6OgdAXWup5oU(XhZO@E1r(s5Jh3RZ uv_Qdj@*^cv3lu>V4p4=HUw$4ou~Z#~$&ZyZg)rqdKT`U_wE2LVgCqbp=RF(% delta 412 zcmZo@U~Fh$6iYG)^Ym4))H7fJ0s#hwp4f%S-_kdV{o-fL*(@mVnV(mjk%8ec6GRp$ zq+rZg-_ORtPzS_bKx_-dnm{ZB#4lMH7_I~HH6Y#r#HB!70>qj?%nHO8SQr>ufj9(+ zje%Gdh{b{U6Eg$DM zwV=cbMGS=lQk0vYlA5Dn3!(MOjmqLv^OAvLMPN2a);R}7Q%O-J!pO`tg}nR{g-oCW ii%Syol2bLn=IJPeCzct7r6wEsf{fR++Pq)QQ4#>S7g?VG 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 8c79c0f75..37ad7edf1 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 @@ -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())