src/transformers/trainer_pt_utils.py/DistributedSamplerWithLoop/__init__
class DistributedSamplerWithLoop:    def __init__(self, dataset, batch_size, **kwargs):
        super().__init__(dataset, **kwargs)
        self.batch_size = batch_size