(self, decay=0.999)
| 56 | self.net_g_ema.eval() |
| 57 | |
| 58 | def model_ema(self, decay=0.999): |
| 59 | net_g = self.get_bare_model(self.net_g) |
| 60 | net_g_params = dict(net_g.named_parameters()) |
| 61 | net_g_ema_params = dict(self.net_g_ema.named_parameters()) |
| 62 | |
| 63 | for k in net_g_ema_params.keys(): |
| 64 | net_g_ema_params[k].data.mul_(decay).add_(net_g_params[k].data, alpha=1 - decay) |
| 65 | |
| 66 | def setup_optimizers(self): |
| 67 | train_opt = self.opt['train'] |
no test coverage detected