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 15:59:04 +09:00
parent bf93d56715
commit e424d4a4ce
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

Binary file not shown.

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

@ -57,12 +57,14 @@ class FairSeqData2VecEncoder(EncoderInterface):
"Please install FairSeq: cd ${MAIN_ROOT}/tools && make fairseq.done" "Please install FairSeq: cd ${MAIN_ROOT}/tools && make fairseq.done"
) )
raise e raise e
'''
if os.path.exists('/home/work/workspace/models/data2vec_model/audio_base_ls.pt'): 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' self.w2v_model_path = '/home/work/workspace/models/data2vec_model/audio_base_ls.pt'
if os.path.exists('/workspace/models/audio_base_ls.pt'): if os.path.exists('/workspace/models/audio_base_ls.pt'):
self.w2v_model_path = '/workspace/models/audio_base_ls.pt' self.w2v_model_path = '/workspace/models/audio_base_ls.pt'
'''
self.w2v_model_path = download_d2v()
self._output_size = output_size self._output_size = output_size
models, _, _ = fairseq.checkpoint_utils.load_model_ensemble_and_task( models, _, _ = fairseq.checkpoint_utils.load_model_ensemble_and_task(