Method reset

datasets/sampler.py:352–353

(self)

        return iter(indices_i)

    def reset(self):
        self.start_index = 0

    def state_dict(self, step) -> dict:
        return {'start_index': step}