Log metrics.
(self, data: dict, step: int, commit: bool = False)
| 18 | self.console_logger = get_logger(__name__) |
| 19 | |
| 20 | def log(self, data: dict, step: int, commit: bool = False) -> None: |
| 21 | """Log metrics.""" |
| 22 | for key in data: |
| 23 | self.logger.add_scalar(key, data[key], step) |
| 24 | self.console_logger.info(f"Step {step}: {data}") |
| 25 | |
| 26 | def close(self) -> None: |
| 27 | self.logger.close() |
no outgoing calls