(self, update_value: torch.Tensor)
| 319 | self.running_average = 0 |
| 320 | |
| 321 | def update(self, update_value: torch.Tensor): |
| 322 | new_average = self.momentum * self.running_average |
| 323 | self.running_average = update_value + new_average |
| 324 | |
| 325 | |
| 326 |
no outgoing calls
no test coverage detected