(self, metrics)
| 29 | self.running_loss[k] = 0.0 |
| 30 | |
| 31 | def push(self, metrics): |
| 32 | self.total_steps += 1 |
| 33 | |
| 34 | for key in metrics: |
| 35 | if key not in self.running_loss: |
| 36 | self.running_loss[key] = 0.0 |
| 37 | |
| 38 | self.running_loss[key] += metrics[key] |
| 39 | |
| 40 | if self.total_steps % self.cfg.sum_freq == self.cfg.sum_freq-1: |
| 41 | self._print_training_status() |
| 42 | self.running_loss = {} |
| 43 | |
| 44 | def write_dict(self, results): |
| 45 | if self.writer is None: |
no test coverage detected