examples/research_projects/seq2seq-distillation/utils.py/SortishSampler/__iter__
class SortishSampler:    def __iter__(self):
        return iter(sortish_sampler_indices(self.data, self.bs, shuffle=self.shuffle))