(self, metrics, epoch=None)
| 328 | self.num_bad_epochs = 0 |
| 329 | |
| 330 | def step(self, metrics, epoch=None): |
| 331 | # convert `metrics` to float, in case it's a zero-dim Tensor |
| 332 | current = float(metrics) |
| 333 | self.last_epoch = epoch |
| 334 | |
| 335 | if self.is_better(current, self.best): |
| 336 | self.best = current |
| 337 | self.num_bad_epochs = 0 |
| 338 | else: |
| 339 | self.num_bad_epochs += 1 |
| 340 | |
| 341 | if self.num_bad_epochs > self.patience: |
| 342 | print('Tracked metric has plateaud') |
| 343 | self._reset() |
| 344 | return True |
| 345 | else: |
| 346 | return False |
| 347 | |
| 348 | def is_better(self, a, best): |
| 349 |
no test coverage detected