From 0207b964263f279ec9b44bc7f77dc1c96d84f024 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 3 Jan 2023 13:41:44 +0900 Subject: [PATCH] from local --- .../.data2vec_audio.py.swp | Bin 45056 -> 45056 bytes .../data2vec_audio.py | 6 ++++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp index 28225c5c7ecf78374778e71dac8f8be906e6ad4a..d5954b67944386ffd8e15691116aec841ab9132b 100644 GIT binary patch delta 543 zcmXBRODM!q6vy#1f3IO?_$$VQ@{fhkFbft4lPMvy@JLZZd6PUMMQMf-8>`C(St&GW zn*KX^?vT=0u~3s@r7SF^eC1c4+gY6MJ@?*FTnoiD=YXfX#cx!XdxYpBZ&7owekkg! z(Je03t`}GW=B%zN9ItkfKF5ELOJ#aTEw$#N_tdV+)@({IO)`D{>yBVp6RA!7?+X0w zNG>u&_A!ZGR6@gDx=0vvXoLq9&|q@XF%GbdF|?uzR(vu|3|F|s1x&jHWrJcJ z1FKkpA8jbXW3tFS!q`R=yl6l@67ZSC?XZVk?4Sp3eDlh2+~NkC=yI@h;_nNZ6Qi*c zO?_-x&G5lbnz9$#R3wt4Tx+>XFHBKBe^HJ>#z&0##hKxLqpr><9-N+9D6UcG5#KMS ConYbs delta 482 zcmXBQ&nv?L7{~Ev+ZSWs*>|m`wZkNtS}{`E_cWw9OmVQ9n;mGoa8YV(HUEIV#DSAy zO<}%VHmMyrcR0sBmJmk-y$GTQe$=_@6ZfiQB}21<7>3b}ibLcK zMdT5MjtE?MY7;4A8)=LnikDWA8|1Nu45r|QhF81D12(XZIeS)QoTlC)@{TuL;sTq{ zF$52Onng}fz!46SMjSyXu)&II6UQNm2@D{NFFt*R3jV)tDoU5l_}H?W9ix7SaeJ!& E0_f>YKL7v# diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py index e85f80260..5cd4af817 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py @@ -145,9 +145,11 @@ class ResidualAdapterModule(nn.Module): super().__init__() - self.adapters = ConvolutionModule(768, 31) def build_adapter(embedding_dim, proj_dim, type_='conv'): - return nn.Sequential( + if type_ == 'conv': + return ConvolutionModule(768, 31) + else: + return nn.Sequential( #nn.LayerNorm(embedding_dim), nn.Linear(embedding_dim, proj_dim), nn.ReLU(),