(train_flag, model_name)
| 45 | |
| 46 | |
| 47 | def initEnv(train_flag, model_name): |
| 48 | cfgs_root = 'cfgs' |
| 49 | cur_cfg = getConfig(cfgs_root, model_name) |
| 50 | |
| 51 | root_dir = cur_cfg['output_root'] |
| 52 | cur_cfg['model_name'] = model_name |
| 53 | version = cur_cfg['output_version'] |
| 54 | work_dir = os.path.join(root_dir, model_name, version) |
| 55 | |
| 56 | backup_name = cur_cfg['backup_name'] |
| 57 | log_name = cur_cfg['log_name'] |
| 58 | |
| 59 | backup_dir = os.path.join(work_dir, backup_name) |
| 60 | log_dir = os.path.join(work_dir, log_name) |
| 61 | |
| 62 | |
| 63 | if train_flag == 1: |
| 64 | safeMakeDirs(backup_dir) |
| 65 | stdout_flag = cur_cfg['train']['stdout'] |
| 66 | setLogging(log_dir, stdout_flag) |
| 67 | |
| 68 | gpus = cur_cfg['train']['gpus'] |
| 69 | os.environ['CUDA_VISIBLE_DEVICES'] = gpus |
| 70 | |
| 71 | cur_cfg['train']['backup_dir'] = backup_dir |
| 72 | elif train_flag == 2: |
| 73 | stdout_flag = cur_cfg['test']['stdout'] |
| 74 | setLogging(log_dir, stdout_flag) |
| 75 | |
| 76 | gpus = cur_cfg['test']['gpus'] |
| 77 | os.environ['CUDA_VISIBLE_DEVICES'] = gpus |
| 78 | else: |
| 79 | gpus = cur_cfg['speed']['gpus'] |
| 80 | os.environ['CUDA_VISIBLE_DEVICES'] = gpus |
| 81 | |
| 82 | ret_cfg = combineConfig(cur_cfg, train_flag) |
| 83 | |
| 84 | return ret_cfg |
| 85 | |
| 86 | |
| 87 | def randomSeeding(seed): |
no test coverage detected