(self)
| 192 | self.train_iter = iter(self.train_dataloader) |
| 193 | |
| 194 | def sample_data_basket(self): |
| 195 | try: |
| 196 | real_image_basket, real_label_basket = next(self.train_iter) |
| 197 | except StopIteration: |
| 198 | self.epoch_counter += 1 |
| 199 | if self.RUN.train and self.DDP: |
| 200 | self.train_dataloader.sampler.set_epoch(self.epoch_counter) |
| 201 | else: |
| 202 | pass |
| 203 | self.train_iter = iter(self.train_dataloader) |
| 204 | real_image_basket, real_label_basket = next(self.train_iter) |
| 205 | |
| 206 | real_image_basket = torch.split(real_image_basket, self.OPTIMIZATION.batch_size) |
| 207 | real_label_basket = torch.split(real_label_basket, self.OPTIMIZATION.batch_size) |
| 208 | return real_image_basket, real_label_basket |
| 209 | |
| 210 | # ----------------------------------------------------------------------------- |
| 211 | # train Discriminator |
no outgoing calls
no test coverage detected