(self, data)
| 83 | raise ValueError('Both pixel and perceptual losses are None.') |
| 84 | |
| 85 | def feed_data(self, data): |
| 86 | self.gt = data['gt'].to(self.device) |
| 87 | self.gt_path = data['gt_path'] |
| 88 | self.lq = data.get('lq', None) |
| 89 | if self.lq is not None: |
| 90 | self.lq = self.lq.to(self.device) |
| 91 | self.lq_path = data.get('lq_path', None) |
| 92 | |
| 93 | def optimize_parameters(self, current_iter): |
| 94 | self.optimizer_g.zero_grad() |