(args=None)
| 28 | |
| 29 | |
| 30 | def args_parser(args=None): |
| 31 | parser = argparse.ArgumentParser("Training process...") |
| 32 | parser.add_argument('--cfg', help='Config file', required=True) |
| 33 | parser.add_argument('--alloc_mem', '-a', help='Pre allocating GPU memory', action='store_true') |
| 34 | return parser.parse_args(args) |
| 35 | |
| 36 | |
| 37 | if __name__=='__main__': |