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 6a092174b..8ea8ee60e 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 b95fd7485..9c1f28bdb 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 @@ -496,6 +496,7 @@ class Data2VecAudioModel(BaseFairseqModel): spk_dir = f'/home/work/workspace/icefall/egs/librispeech/ASR/conv_feat/{sid}' if not os.path.isdir(spk_dir): os.mkdir(spk_dir) + conv_feat_all = torch.tensor([]).to(features.device) length = 0 for i in range(padding_mask.size()[0]):