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

Add a note about the CUDA OOM error. (#94)

Browse Source 
* Add a note about the CUDA OOM error.

Some users consider this kind of OOM as an error during decoding,
but actually it is not. This pull request clarifies that.

* Fix style issues.
This commit is contained in:
Fangjun Kuang 2021-10-29 12:17:56 +08:00 committed by GitHub
parent 8cb7f712e4
commit 12d647d899
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
icefall/decode.py
View File

@ -732,6 +732,12 @@ def rescore_with_whole_lattice(
logging.info(
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
# to avoid OOM. You may need to fine tune it.