| 12 | |
| 13 | |
| 14 | def make_parser(): |
| 15 | parser = argparse.ArgumentParser("YOLOX train parser") |
| 16 | parser.add_argument("-expn", "--experiment-name", type=str, default=None) |
| 17 | parser.add_argument("-n", "--name", type=str, default=None, help="model name") |
| 18 | |
| 19 | # distributed |
| 20 | parser.add_argument( |
| 21 | "--dist-backend", default="nccl", type=str, help="distributed backend" |
| 22 | ) |
| 23 | parser.add_argument( |
| 24 | "--dist-url", |
| 25 | default=None, |
| 26 | type=str, |
| 27 | help="url used to set up distributed training", |
| 28 | ) |
| 29 | parser.add_argument("-b", "--batch-size", type=int, default=64, help="batch size") |
| 30 | parser.add_argument( |
| 31 | "-d", "--devices", default=None, type=int, help="device for training" |
| 32 | ) |
| 33 | parser.add_argument( |
| 34 | "--local_rank", default=0, type=int, help="local rank for dist training" |
| 35 | ) |
| 36 | parser.add_argument( |
| 37 | "-f", |
| 38 | "--exp_file", |
| 39 | default=None, |
| 40 | type=str, |
| 41 | help="plz input your expriment description file", |
| 42 | ) |
| 43 | parser.add_argument( |
| 44 | "--resume", default=False, action="store_true", help="resume training" |
| 45 | ) |
| 46 | parser.add_argument("-c", "--ckpt", default=None, type=str, help="checkpoint file") |
| 47 | parser.add_argument( |
| 48 | "-e", |
| 49 | "--start_epoch", |
| 50 | default=None, |
| 51 | type=int, |
| 52 | help="resume training start epoch", |
| 53 | ) |
| 54 | parser.add_argument( |
| 55 | "--num_machines", default=1, type=int, help="num of node for training" |
| 56 | ) |
| 57 | parser.add_argument( |
| 58 | "--machine_rank", default=0, type=int, help="node rank for multi-node training" |
| 59 | ) |
| 60 | parser.add_argument( |
| 61 | "--fp16", |
| 62 | dest="fp16", |
| 63 | default=True, |
| 64 | action="store_true", |
| 65 | help="Adopting mix precision training.", |
| 66 | ) |
| 67 | parser.add_argument( |
| 68 | "-o", |
| 69 | "--occupy", |
| 70 | dest="occupy", |
| 71 | default=False, |