(self, scalar_dict, step, prefix=None)
| 21 | self.writer = None |
| 22 | |
| 23 | def log_scalars(self, scalar_dict, step, prefix=None): |
| 24 | if self.writer is None: |
| 25 | return |
| 26 | for k in scalar_dict: |
| 27 | v = scalar_dict[k] |
| 28 | if isinstance(v, torch.Tensor): |
| 29 | v = v.detach().cpu().item() |
| 30 | if prefix is not None: |
| 31 | k = prefix + '_' + k |
| 32 | self.writer.add_scalar(k, v, step) |
| 33 | return |
| 34 | |
| 35 | def log_messages(self, message: str): |
| 36 | if self.txt_writer is not None: |