From a895d5b2f94ccacf77a8ef3c84e5ebd7ed6f6def Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Tue, 3 Jan 2023 13:21:48 +0900 Subject: [PATCH] from local --- .../.data2vec_audio.py.swp | Bin 40960 -> 40960 bytes .../data2vec_audio.py | 4 +++- 2 files changed, 3 insertions(+), 1 deletion(-) 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 4b53c9534a8cdd95bceda5317f82c039b9662c1b..7b3a3ea9825108db0b5efbfefafe433cb413cb04 100644 GIT binary patch delta 455 zcmXZYPbh-{7{~Evf6SQi60sb-^QVkW@7v0sTr8)R9h9_kkv|7ndyy>{Z6|Y}=uJXG z%FRJN(3YD!-MY%{)9XpaE}XYA_oH#(4oPLveU9T z(+)GRroQaboQ+m-&qAva-4?LMiFHLN0r_o#G*0LMLLSRlLJBWk(mosv zKzIsCkGMb~G%M{;Y|LOBX_)xsryqRc22IpZMjo5UA|3R6fre3OA6cF?QVmx=s$+lg CdOzy` 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 81cacba88..b384f6d40 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 @@ -39,7 +39,9 @@ logger = logging.getLogger().setLevel(logging.INFO) class TransformerEncoderAdapter(TransformerEncoder): def __init__(self, args: Wav2Vec2Config): super().__init__(args) - self.adapters = ResidualAdapterModule() + #self.adapters = ResidualAdapterModule() + self.adapters = ConvolutionModule(768, 31) + for p in self.adapters.parameters(): #p.data = nn.Parameter(torch.zeros(p.size()).to('cuda')) p.data = nn.Parameter(torch.randn(p.size()).to('cuda')/50.)