(self, decay=0.999)
| 74 | self.best_metric_results[dataset_name][metric]['iter'] = current_iter |
| 75 | |
| 76 | def model_ema(self, decay=0.999): |
| 77 | net_g = self.get_bare_model(self.net_g) |
| 78 | |
| 79 | net_g_params = dict(net_g.named_parameters()) |
| 80 | net_g_ema_params = dict(self.net_g_ema.named_parameters()) |
| 81 | |
| 82 | for k in net_g_ema_params.keys(): |
| 83 | net_g_ema_params[k].data.mul_(decay).add_(net_g_params[k].data, alpha=1 - decay) |
| 84 | |
| 85 | def get_current_log(self): |
| 86 | return self.log_dict |
no test coverage detected