mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-20 16:00:28 +09:00
parent 441020813d
commit 7df54a838c
3 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/.run_v3.sh.swp
View File

Binary file not shown.

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_encoder.py.swp
View File

Binary file not shown.

21
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_encoder.py
View File

@ -20,6 +20,15 @@ from encoder_interface import EncoderInterface
from torch import Tensor, nn
from icefall.utils import make_pad_mask, subsequent_chunk_mask
try:
import fairseq
from data2vec_audio import *
except Exception as e:
print("Error: FairSeq is not properly installed.")
print(
"Please install FairSeq: cd ${MAIN_ROOT}/tools && make fairseq.done"
)
raise e
class FairSeqData2VecEncoder(EncoderInterface):
@ -47,17 +56,7 @@ class FairSeqData2VecEncoder(EncoderInterface):
assert check_argument_types()
super().__init__()
if w2v_url != "":
try:
import fairseq
from data2vec_audio import *
except Exception as e:
print("Error: FairSeq is not properly installed.")
print(
"Please install FairSeq: cd ${MAIN_ROOT}/tools && make fairseq.done"
)
raise e
'''
if os.path.exists('/home/work/workspace/models/data2vec_model/audio_base_ls.pt'):
self.w2v_model_path = '/home/work/workspace/models/data2vec_model/audio_base_ls.pt'