(self, mu=0.999)
| 5 | |
| 6 | class EMAHelper(object): |
| 7 | def __init__(self, mu=0.999): |
| 8 | self.mu = mu |
| 9 | self.shadow = {} |
| 10 | |
| 11 | def register(self, module): |
| 12 | if isinstance(module, nn.DataParallel): |
nothing calls this directly
no outgoing calls
no test coverage detected