| 10 | |
| 11 | |
| 12 | def parse_args(): |
| 13 | parser = argparse.ArgumentParser(description='Training') |
| 14 | parser.add_argument('--config', required=True) |
| 15 | parser.add_argument('--name', required=True) |
| 16 | parser.add_argument('--checkpoints_dir', default='result', help='Dir for saving logs and models.') |
| 17 | parser.add_argument('--seed', type=int, default=0, help='Random seed.') |
| 18 | parser.add_argument('--which_iter', type=int, default=None) |
| 19 | parser.add_argument('--no_resume', action='store_true') |
| 20 | parser.add_argument('--local_rank', type=int, default=0) |
| 21 | parser.add_argument('--single_gpu', action='store_true') |
| 22 | parser.add_argument('--debug', action='store_true') |
| 23 | args = parser.parse_args() |
| 24 | return args |
| 25 | |
| 26 | |
| 27 | def main(): |