(self)
| 71 | self.setup_schedulers() |
| 72 | |
| 73 | def setup_optimizers(self): |
| 74 | train_opt = self.opt['train'] |
| 75 | optim_params = [] |
| 76 | for k, v in self.net_g.named_parameters(): |
| 77 | if v.requires_grad: |
| 78 | optim_params.append(v) |
| 79 | else: |
| 80 | logger = get_root_logger() |
| 81 | logger.warning(f'Params {k} will not be optimized.') |
| 82 | |
| 83 | optim_type = train_opt['optim_g'].pop('type') |
| 84 | self.optimizer_g = self.get_optimizer(optim_type, optim_params, **train_opt['optim_g']) |
| 85 | self.optimizers.append(self.optimizer_g) |
| 86 | |
| 87 | def feed_data(self, data): |
| 88 | self.lq = data['lq'].to(self.device) |
no test coverage detected