(self, dataset, num_replicas, rank, ratio=1)
| 19 | """ |
| 20 | |
| 21 | def __init__(self, dataset, num_replicas, rank, ratio=1): |
| 22 | self.dataset = dataset |
| 23 | self.num_replicas = num_replicas |
| 24 | self.rank = rank |
| 25 | self.epoch = 0 |
| 26 | print(self.dataset,ratio,self.num_replicas) |
| 27 | self.num_samples = math.ceil(len(self.dataset) * ratio / self.num_replicas) |
| 28 | self.total_size = self.num_samples * self.num_replicas |
| 29 | |
| 30 | def __iter__(self): |
| 31 | # deterministically shuffle based on epoch |
nothing calls this directly
no outgoing calls
no test coverage detected