(self, model)
| 476 | ema_v.copy_(update_fn(ema_v, model_v)) |
| 477 | |
| 478 | def update(self, model): |
| 479 | self._update(model, |
| 480 | update_fn=lambda e, m: self.decay * e + |
| 481 | (1. - self.decay) * m) |
| 482 | |
| 483 | def set(self, model): |
| 484 | self._update(model, update_fn=lambda e, m: m) |