(self, args, for_eva=False)
| 15 | |
| 16 | class BasicLogger: |
| 17 | def __init__(self, args, for_eva=False) -> None: |
| 18 | self.args = args |
| 19 | self.log_dir = osp.join( |
| 20 | args.log_dir, args.exp_name, self.get_random_time_str()) |
| 21 | self.img_dir = osp.join(self.log_dir, "imgs") |
| 22 | self.mesh_dir = osp.join(self.log_dir, "mesh") |
| 23 | self.ckpt_dir = osp.join(self.log_dir, "ckpt") |
| 24 | self.backup_dir = osp.join(self.log_dir, "bak") |
| 25 | self.misc_dir = osp.join(self.log_dir, "misc") |
| 26 | self.for_eva = for_eva |
| 27 | if not for_eva: |
| 28 | os.makedirs(self.img_dir) |
| 29 | os.makedirs(self.ckpt_dir) |
| 30 | os.makedirs(self.mesh_dir) |
| 31 | os.makedirs(self.misc_dir) |
| 32 | os.makedirs(self.backup_dir) |
| 33 | |
| 34 | self.log_config(args) |
| 35 | |
| 36 | def get_random_time_str(self): |
| 37 | return datetime.strftime(datetime.now(), "%Y-%m-%d-%H-%M-%S") |
nothing calls this directly
no test coverage detected