From 354601b35067e2bf7f5aab4c6a56b6f23931c422 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Mon, 26 Dec 2022 14:42:39 +0900 Subject: [PATCH] from local --- .../.data2vec_audio.py.swp | Bin 45056 -> 45056 bytes .../data2vec_audio.py | 4 +--- 2 files changed, 1 insertion(+), 3 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 8a737e28464f80ae43428a69133d07cceaf8f2dd..603fd4ff488d3d3099eb6eecf1f60849cc29aef0 100644 GIT binary patch delta 220 zcmZp8z|`=7Ni4}A%+puFQqO<^2m}}y>`hiCn>^SkCMCd_v)NGKay@SrBLjmOGekCL zv)}}O#`+#s28L1~E&<{|AeI4QX(0Z?!octhh%W%~c_7{j#EXG=5fC>5aWW7m0kIbl z>jNUn_f#E3-9|PhIKs*bGX996O5a$AM4iI|-u`Uqn z0I>)V{{$L+4T!G-@g^Xi3B)BpY!1XiKn$|_KNFBffX(+Ox-)K$ne4!1UQm>oSE5l7 zuUDK|m8zks$)JD+D&iGv6^c`H()1Ej5(`RFi;6WWG#Ms8_tuzPtrEFeZrUSB0J9c1 A2><{9 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 f20ca6260..33010e3ff 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 @@ -95,9 +95,7 @@ class TransformerEncoderAdapter(TransformerEncoder): x, (z, lr) = layer( x, self_attn_padding_mask=padding_mask, need_weights=False, ) - print(x.size()) - x_ = self.adapters(x) - print(x_.size()) + x = self.adapters(x, layer_id=i) exit() if i >= min_layer: