(self, model, scheduler, cfg)
| 3 | |
| 4 | class Logger: |
| 5 | def __init__(self, model, scheduler, cfg): |
| 6 | self.model = model |
| 7 | self.scheduler = scheduler |
| 8 | self.total_steps = 0 |
| 9 | self.running_loss = {} |
| 10 | self.writer = None |
| 11 | self.cfg = cfg |
| 12 | |
| 13 | def _print_training_status(self): |
| 14 | metrics_data = [self.running_loss[k]/self.cfg.sum_freq for k in sorted(self.running_loss.keys())] |
nothing calls this directly
no outgoing calls
no test coverage detected