This commit is contained in:
Bailey Hirota 2025-07-01 21:04:18 +09:00
parent d8cb41f4f6
commit 55d0664339

View File

@ -272,8 +272,8 @@ class MultiDatasetAsrDataModule:
max_duration=self.args.max_duration,
shuffle=self.args.shuffle,
num_buckets=self.args.num_buckets,
buffer_size=self.args.num_buckers * 2000,
shuffle_buffer_size=self.args.num_buckers * 5000,
buffer_size=self.args.num_buckets * 2000,
shuffle_buffer_size=self.args.num_buckets * 5000,
drop_last=self.args.drop_last,
)
else: