mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fixes

Browse Source
This commit is contained in:
Fangjun Kuang 2022-07-25 21:10:27 +08:00
parent 90dc5772ec
commit c1fc004df5
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless3/jit_decode.py
View File

@ -90,7 +90,7 @@ def main():
model = torch.jit.load(params.nn_model_filename)
device = torch.device("cpu")
if torch.cuda.is_available() and hasattr(
if torch.cuda.is_available() and not hasattr(
model.simple_lm_proj, "_packed_params"
):
device = torch.device("cuda", 0)
@ -105,6 +105,8 @@ def main():
params.unk_id = sp.piece_to_id("<unk>")
params.vocab_size = sp.get_piece_size()
params.suffix = "jit"
model.to(device)
model.device = device
model.unk_id = params.unk_id