mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use mode.eval() in decoding.

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

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

@ -88,6 +88,7 @@ def main():
logging.info("Decoding started") logging.info("Decoding started")
model = torch.jit.load(params.nn_model_filename) model = torch.jit.load(params.nn_model_filename)
model.eval()
device = torch.device("cpu") device = torch.device("cpu")
if torch.cuda.is_available() and not hasattr( if torch.cuda.is_available() and not hasattr(