(self)
| 12 | self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') |
| 13 | |
| 14 | def analyze(self): |
| 15 | name_list = [hook.name for hook in self.hook_list] |
| 16 | np.random.shuffle(name_list) |
| 17 | |
| 18 | save_folder = f"options/cond/{self.opt['name']}" |
| 19 | os.makedirs(save_folder, exist_ok=True) |
| 20 | np.savetxt(os.path.join(save_folder, 'filter_name.txt'), name_list, delimiter=',', fmt='%s') |
| 21 | |
| 22 | def main(): |
| 23 | root_path = osp.abspath(osp.join(__file__, osp.pardir, osp.pardir)) |