Method reset

datasets/sampler.py:50–51

(self)

        return self.num_samples - self.start_index

    def reset(self) -> None:
        self.start_index = 0

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