mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
auto frontend
This commit is contained in:
parent
8ba12df6f7
commit
870fa4a37a
@ -417,7 +417,7 @@ class CustomDistributedBufferDynamicBatchSampler(DistributedSampler):
|
||||
rank_batches[i % self.num_replicas].append(batch)
|
||||
|
||||
# Assign all batches for the current rank directly
|
||||
final_batches = rank_batches[self.rank][self.start_step :]
|
||||
final_batches = rank_batches[self.rank] # [self.start_step :]
|
||||
self.batch_num = len(final_batches)
|
||||
|
||||
logging.info(
|
||||
|
||||
Loading…
Reference in New Issue
Block a user