(self, module)
| 25 | self.mu) * param.data + self.mu * self.shadow[name].data |
| 26 | |
| 27 | def ema(self, module): |
| 28 | if isinstance(module, nn.DataParallel): |
| 29 | module = module.module |
| 30 | for name, param in module.named_parameters(): |
| 31 | if param.requires_grad: |
| 32 | param.data.copy_(self.shadow[name].data) |
| 33 | |
| 34 | def ema_copy(self, module): |
| 35 | if isinstance(module, nn.DataParallel): |