|
7 | 7 |
|
8 | 8 |
|
9 | 9 | def parse(parser): |
10 | | - parser.add_argument('--conf', '-c', help='path to config file') |
11 | 10 | parser.add_argument('--path', '-p', help='path to model file') |
| 11 | + parser.add_argument('--conf', '-c', default='', help='path to config file') |
12 | 12 | parser.add_argument('--device', '-d', default='-1', help='ID of GPU to use') |
13 | 13 | parser.add_argument('--seed', '-s', default=1, type=int, help='seed for generating random numbers') |
14 | 14 | parser.add_argument('--threads', '-t', default=16, type=int, help='max num of threads') |
15 | 15 | parser.add_argument('--batch-size', default=5000, type=int, help='batch size') |
16 | | - parser.add_argument('--local_rank', default=-1, type=int, help='node rank for distributed training') |
17 | | - parser.add_argument('--epochs', default=5000, type=int, help='epochs') |
| 16 | + parser.add_argument("--local_rank", type=int, default=-1, help='node rank for distributed training') |
18 | 17 | args, unknown = parser.parse_known_args() |
19 | | - args, _ = parser.parse_known_args(unknown, args) |
20 | | - args = Config(**vars(args)) |
| 18 | + args, unknown = parser.parse_known_args(unknown, args) |
| 19 | + args = Config.load(**vars(args), unknown=unknown) |
21 | 20 | Parser = args.pop('Parser') |
22 | 21 |
|
23 | 22 | torch.set_num_threads(args.threads) |
|
0 commit comments