(self)
| 128 | self.net_g.train() |
| 129 | |
| 130 | def get_current_visuals(self): |
| 131 | out_dict = OrderedDict() |
| 132 | out_dict['lq'] = self.lq.detach().cpu() |
| 133 | out_dict['result'] = self.output.detach().cpu() |
| 134 | if hasattr(self, 'gt'): |
| 135 | out_dict['gt'] = self.gt.detach().cpu() |
| 136 | return out_dict |
| 137 | |
| 138 | |
| 139 | @master_only |