| 13 | |
| 14 | |
| 15 | def parse_args(): |
| 16 | parser = argparse.ArgumentParser(description='Run case analyzer') |
| 17 | parser.add_argument('config', help='Train config file path') |
| 18 | parser.add_argument( |
| 19 | '-f', |
| 20 | '--force', |
| 21 | help='Force to run the task even if the results already exist', |
| 22 | action='store_true', |
| 23 | default=False) |
| 24 | parser.add_argument('-w', |
| 25 | '--work-dir', |
| 26 | help='Work path, all the outputs will be ' |
| 27 | 'saved in this path, including the slurm logs, ' |
| 28 | 'the evaluation results, the summary results, etc.' |
| 29 | 'If not specified, the work_dir will be set to ' |
| 30 | './outputs/default.', |
| 31 | default=None, |
| 32 | type=str) |
| 33 | args = parser.parse_args() |
| 34 | return args |
| 35 | |
| 36 | |
| 37 | class BadcaseShower: |