Update egs/mls_english/ASR/local/utils/asr_datamodule.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
This commit is contained in:
Kinan Martin 2025-09-02 18:07:06 +09:00 committed by GitHub
parent f64a706191
commit ef7664e7cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -75,7 +75,7 @@ class MLSEnglishHFAsrDataModule:
) )
group.add_argument( group.add_argument(
"--max-duration", "--max-duration",
type=int, type=float,
default=200.0, default=200.0,
help="Maximum pooled recordings duration (seconds) in a " help="Maximum pooled recordings duration (seconds) in a "
"single batch. You can reduce it if it causes CUDA OOM.", "single batch. You can reduce it if it causes CUDA OOM.",