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

Add a note about the CUDA OOM error.

Browse Source 
Some users consider this kind of OOM as an error during decoding,
but actually it is not. This pull request clarifies that.
This commit is contained in:
Fangjun Kuang 2021-10-27 16:59:47 +08:00
parent 8cb7f712e4
commit a7cf41d3ac
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
icefall/decode.py
View File

@ -732,6 +732,11 @@ def rescore_with_whole_lattice(
logging.info( logging.info(
f"num_arcs before pruning: {inv_lattice.arcs.num_elements()}" f"num_arcs before pruning: {inv_lattice.arcs.num_elements()}"
) )
logging.info("This OOM is not an error. You can ignore it. "
"If your model does not converge well, or --max-duration "
"is too large, or the input sound file is difficult to "
"decode, you will meet this exception."
)
# NOTE(fangjun): The choice of the threshold 1e-9 is arbitrary here # NOTE(fangjun): The choice of the threshold 1e-9 is arbitrary here
# to avoid OOM. You may need to fine tune it. # to avoid OOM. You may need to fine tune it.