auto frontend

This commit is contained in:
游雁 2024-06-07 04:05:30 +08:00
parent 8ba12df6f7
commit 870fa4a37a

View File

@ -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(