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-30 11:17:59 +09:00
parent 82bce0fbb0
commit e9f3b3efa7
3 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

Binary file not shown.

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

Binary file not shown.

3
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/decode.py
View File

@ -663,6 +663,7 @@ def main():
logging.info("About to create model") logging.info("About to create model")
model = get_transducer_model(params) model = get_transducer_model(params)
print(model)
if params.model_name: if params.model_name:
load_checkpoint(f"{params.exp_dir}/{params.model_name}", model) load_checkpoint(f"{params.exp_dir}/{params.model_name}", model)
@ -757,7 +758,7 @@ def main():
torch.load(lg_filename, map_location=device) torch.load(lg_filename, map_location=device)
) )
decoding_graph.scores *= params.ngram_lm_scale decoding_graph.scores *= params.ngram_lm_scale
else: else
word_table = None word_table = None
decoding_graph = k2.trivial_graph(params.vocab_size - 1, device=device) decoding_graph = k2.trivial_graph(params.vocab_size - 1, device=device)
else: else: