(self, exp_name)
| 117 | add_pypath(self.data_dir) |
| 118 | |
| 119 | def prepare_dirs(self, exp_name): |
| 120 | time_str = datetime.datetime.now().strftime('%Y%m%d_%H%M%S') |
| 121 | self.output_dir = osp.join(self.root_dir, f'{exp_name}_{time_str}') |
| 122 | self.model_dir = osp.join(self.output_dir, 'model_dump') |
| 123 | self.vis_dir = osp.join(self.output_dir, 'vis') |
| 124 | self.log_dir = osp.join(self.output_dir, 'log') |
| 125 | self.code_dir = osp.join(self.output_dir, 'code') |
| 126 | self.result_dir = osp.join(self.output_dir.split('/')[:-1]) |
| 127 | from util.dir import make_folder |
| 128 | make_folder(self.model_dir) |
| 129 | make_folder(self.vis_dir) |
| 130 | make_folder(self.log_dir) |
| 131 | make_folder(self.code_dir) |
| 132 | make_folder(self.result_dir) |
| 133 | |
| 134 | ## copy some code to log dir as a backup |
| 135 | copy_files = [ |
| 136 | 'main/train.py', 'main/test.py', 'common/base.py', 'main/OSX.py', |
| 137 | 'common/nets', 'main/OSX_WoDecoder.py', 'data/dataset.py', |
| 138 | 'data/MSCOCO/MSCOCO.py', 'data/AGORA/AGORA.py' |
| 139 | ] |
| 140 | for file in copy_files: |
| 141 | os.system(f'cp -r {self.root_dir}/{file} {self.code_dir}') |
| 142 | |
| 143 | def update_test_config(self, testset, agora_benchmark, shapy_eval_split, |
| 144 | pretrained_model_path, use_cache): |
no test coverage detected