Method set_num_samples

datasets/sampler.py:108–110

(self)

        self.epoch = epoch

    def set_num_samples(self):
        self.num_samples = int(math.ceil(len(self.dataset) * 1.0 / self.num_replicas))
        self.total_size = self.num_samples * self.num_replicas