(self, dataset, batch_size, k, num_replicas, rank, seed=0)
| 117 | |
| 118 | class DistributedKRepeatSampler(Sampler): |
| 119 | def __init__(self, dataset, batch_size, k, num_replicas, rank, seed=0): |
| 120 | self.dataset = dataset |
| 121 | self.batch_size = batch_size |
| 122 | self.k = k |
| 123 | self.num_replicas = num_replicas |
| 124 | self.rank = rank |
| 125 | self.seed = seed |
| 126 | |
| 127 | self.total_samples = self.num_replicas * self.batch_size |
| 128 | assert ( |
| 129 | self.total_samples % self.k == 0 |
| 130 | ), f"k can not div n*b, k{k}-num_replicas{num_replicas}-batch_size{batch_size}" |
| 131 | self.m = self.total_samples // self.k |
| 132 | self.epoch = 0 |
| 133 | |
| 134 | def __iter__(self): |
| 135 | while True: |
nothing calls this directly
no outgoing calls
no test coverage detected