| 52 | |
| 53 | |
| 54 | class Logger: |
| 55 | def __init__(self, log_path=None, reopen_to_flush=False): |
| 56 | self.log_file = None |
| 57 | self.reopen_to_flush = reopen_to_flush |
| 58 | if log_path is not None: |
| 59 | os.makedirs(os.path.dirname(log_path), exist_ok=True) |
| 60 | self.log_file = open(log_path, 'a+') |
| 61 | |
| 62 | def log(self, msg): |
| 63 | formatted = '[{}] {}'.format( |
| 64 | datetime.datetime.now().replace(microsecond=0).isoformat(), |
| 65 | msg) |
| 66 | print(formatted) |
| 67 | if self.log_file: |
| 68 | self.log_file.write(formatted + '\n') |
| 69 | if self.reopen_to_flush: |
| 70 | log_path = self.log_file.name |
| 71 | self.log_file.close() |
| 72 | self.log_file = open(log_path, 'a+') |
| 73 | else: |
| 74 | self.log_file.flush() |
| 75 | |
| 76 | class Trainer: |
| 77 | def __init__(self, logger, config): |