speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix python runtime

Browse Source
This commit is contained in:
游雁 2024-07-25 09:55:12 +08:00
parent 4b46588c45
commit 4fe50f431b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/datasets/audio_datasets/samplers.py
View File

@ -368,7 +368,7 @@ class CustomDistributedBufferDynamicBatchSampler(DistributedSampler):
self.batch_size_sample_max = kwargs.get("batch_size_sample_max", 200)
self.start_step = start_step
self.batch_num = 1
self.batch_size_scale_threshold = kwargs.get("batch_size_scale_threshold", 40.0)
self.batch_size_scale_threshold = kwargs.get("batch_size_scale_threshold", 4000.0)
if self.start_step > 0:
logging.info(f"Warning, start_step > 0, dataloader start from step: {self.start_step}")
# super().__init__(