(self)
| 540 | return self.best_all.update(new_res, epoch) |
| 541 | |
| 542 | def summary(self): |
| 543 | if not self.use_ema: |
| 544 | return self.best_all.summary() |
| 545 | |
| 546 | res = {} |
| 547 | res.update({f'all_{k}': v for k, v in self.best_all.summary().items()}) |
| 548 | res.update({ |
| 549 | f'regular_{k}': v |
| 550 | for k, v in self.best_regular.summary().items() |
| 551 | }) |
| 552 | res.update({f'ema_{k}': v for k, v in self.best_ema.summary().items()}) |
| 553 | return res |
| 554 | |
| 555 | def __repr__(self) -> str: |
| 556 | return json.dumps(self.summary(), indent=2) |