()
| 61 | |
| 62 | |
| 63 | def get_arguments(): |
| 64 | parser = argparse.ArgumentParser() |
| 65 | parser.add_argument('case', type=str, help='experiment case name') |
| 66 | parser.add_argument('-ckpt', help='checkpoint to load model.') |
| 67 | parser.add_argument('-gpu', help='comma separated list of GPU(s) to use.') |
| 68 | arguments = parser.parse_args() |
| 69 | return arguments |
| 70 | |
| 71 | if __name__ == '__main__': |
| 72 | args = get_arguments() |