from local
This commit is contained in:
parent
e9f3b3efa7
commit
e4e098470b
Binary file not shown.
@ -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)
|
print(model)
|
||||||
|
|
||||||
if params.model_name:
|
if params.model_name:
|
||||||
@ -758,7 +759,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:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user