(self, log_dir=None, accelerator=None)
| 7 | class Logger(object): |
| 8 | |
| 9 | def __init__(self, log_dir=None, accelerator=None) -> None: |
| 10 | self.log_dir = log_dir |
| 11 | self.accelerator = accelerator |
| 12 | |
| 13 | if self.log_dir is not None: |
| 14 | self.txt_writer = open(os.path.join(self.log_dir, 'logger.log'), 'a') |
| 15 | else: |
| 16 | self.txt_writer = None |
| 17 | |
| 18 | if SummaryWriter is not None and self.accelerator.is_main_process: |
| 19 | self.writer = SummaryWriter(self.log_dir) |
| 20 | else: |
| 21 | self.writer = None |
| 22 | |
| 23 | def log_scalars(self, scalar_dict, step, prefix=None): |
| 24 | if self.writer is None: |
nothing calls this directly
no outgoing calls
no test coverage detected