(self)
| 176 | return batch_list |
| 177 | |
| 178 | def __iter__(self): |
| 179 | if self.shuffle or self.is_training: |
| 180 | random.seed(self.epoch) |
| 181 | self.epoch += 1 |
| 182 | self.batch_list = self.create_batch() |
| 183 | random.shuffle(self.batchs_in_one_epoch_id) |
| 184 | for batch_tuple_id in self.batchs_in_one_epoch_id: |
| 185 | yield self.batch_list[batch_tuple_id] |
| 186 | |
| 187 | def set_epoch(self, epoch: int): |
| 188 | self.epoch = epoch |
nothing calls this directly
no test coverage detected