1
2
3
4
5
6
7
8
9
10
11
12
| import argparse
parser = argparse.ArgumentParser()
parser.add_argument('config', metavar='CFG',
help='path to configuration')
parser.add_argument('-t', '--threads', type=int, default=4,
help="how many GPUs are avaliable on this machine")
parser.add_argument('-num', '--nums_epoch', default=200, type=int,
help='nums of epochs')
parser.add_argument('-save_dir', '--save_dir', default="tmp", type=str,
help='save_dir')
parser.add_argument('-gpus', '--gpus', nargs='+', type=int,
help='--gpus 1234 2345 3456 4567 ')
|