|
|
@@ -416,7 +416,7 @@ class BucketedBatchSampler(ReproducibleBatchSampler): |
|
|
|
@property |
|
|
|
def batch_idx_in_epoch(self): |
|
|
|
if self.drop_last: |
|
|
|
return len(self.dataset) // self.batch_size - (len(self.dataset) - self.num_consumed_samples) // self.batch_size |
|
|
|
return len(self.dataset) // self.num_replicas // self.batch_size - self.num_left_samples // self.batch_size |
|
|
|
else: |
|
|
|
return (len(self.dataset) + self.batch_size - 1) // self.batch_size - \ |
|
|
|
(len(self.dataset) - self.num_consumed_samples + self.batch_size - 1) // self.batch_size |
|
|
|
return (len(self.dataset) // self.num_replicas + self.batch_size - 1) // self.batch_size - \ |
|
|
|
(self.num_left_samples + self.batch_size - 1) // self.batch_size |