Use --bucketing-sampler by default.

This commit is contained in:
Fangjun Kuang 2021-09-09 13:45:12 +08:00
parent f1155b6cff
commit 9288f2e9f3

View File

@ -82,14 +82,14 @@ class LibriSpeechAsrDataModule(DataModule):
group.add_argument( group.add_argument(
"--max-duration", "--max-duration",
type=int, type=int,
default=500.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.",
) )
group.add_argument( group.add_argument(
"--bucketing-sampler", "--bucketing-sampler",
type=str2bool, type=str2bool,
default=False, default=True,
help="When enabled, the batches will come from buckets of " help="When enabled, the batches will come from buckets of "
"similar duration (saves padding frames).", "similar duration (saves padding frames).",
) )