(self, value_dict, epoch)
| 77 | self.cfg = cfg |
| 78 | |
| 79 | def record(self, value_dict, epoch): |
| 80 | for key in value_dict: |
| 81 | if isinstance(value_dict[key], AVGMeter): |
| 82 | self.writer.add_scalar(key, value_dict[key].agg(), epoch) |
| 83 | else: |
| 84 | self.writer.add_scalar(key, value_dict[key], epoch) |
| 85 | |
| 86 | def close(self): |
| 87 | self.writer.close() |