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 ffadd5bab..8f9332057 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp differ 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 c9d53b596..46a9981af 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 @@ -40,7 +40,7 @@ class TransformerEncoderAdapter(TransformerEncoder): super().__init__(args) self.adapters = ResidualAdapterModule() for p in self.adapters.parameters(): - p.data = torch.nn.parameter.Parameter(torch.zeros(p.size()).to('cuda')) + p.data = nn.parameter.Parameter(torch.zeros(p.size()).to('cuda')) print(p) def forward(self, x, padding_mask=None, layer=None, tgt_layer=None):