src/transformers/trainer_pt_utils.py/IterableDatasetShard/set_epoch
class IterableDatasetShard:    def set_epoch(self, epoch):
        self.epoch = epoch
        if hasattr(self.dataset, "set_epoch"):
            self.dataset.set_epoch(epoch)