(self, args)
| 93 | |
| 94 | class Logger(object): |
| 95 | def __init__(self, args): |
| 96 | self.args = args |
| 97 | self.save_dir = args.save_dir |
| 98 | self.is_primary = is_primary() |
| 99 | |
| 100 | if self.is_primary: |
| 101 | os.makedirs(self.save_dir, exist_ok=True) |
| 102 | |
| 103 | # save the args and config |
| 104 | self.config_dir = os.path.join(self.save_dir, 'configs') |
| 105 | os.makedirs(self.config_dir, exist_ok=True) |
| 106 | file_name = os.path.join(self.config_dir, 'args.txt') |
| 107 | write_args(args, file_name) |
| 108 | |
| 109 | log_dir = os.path.join(self.save_dir, 'logs') |
| 110 | if not os.path.exists(log_dir): |
| 111 | os.makedirs(log_dir, exist_ok=True) |
| 112 | self.text_writer = open(os.path.join(log_dir, 'log.txt'), |
| 113 | 'a') # 'w') |
| 114 | if args.tensorboard: |
| 115 | self.log_info('using tensorboard') |
| 116 | self.tb_writer = torch.utils.tensorboard.SummaryWriter( |
| 117 | log_dir=log_dir |
| 118 | ) # tensorboard.SummaryWriter(log_dir=log_dir) |
| 119 | else: |
| 120 | self.tb_writer = None |
| 121 | |
| 122 | def save_config(self, config): |
| 123 | if self.is_primary: |
nothing calls this directly
no test coverage detected