| 20 | |
| 21 | |
| 22 | def make_parser(): |
| 23 | parser = argparse.ArgumentParser("YOLOX Eval") |
| 24 | parser.add_argument("-expn", "--experiment-name", type=str, default=None) |
| 25 | parser.add_argument("-n", "--name", type=str, default=None, help="model name") |
| 26 | |
| 27 | # distributed |
| 28 | parser.add_argument( |
| 29 | "--dist-backend", default="nccl", type=str, help="distributed backend" |
| 30 | ) |
| 31 | parser.add_argument( |
| 32 | "--dist-url", |
| 33 | default=None, |
| 34 | type=str, |
| 35 | help="url used to set up distributed training", |
| 36 | ) |
| 37 | parser.add_argument("-b", "--batch-size", type=int, default=64, help="batch size") |
| 38 | parser.add_argument( |
| 39 | "-d", "--devices", default=None, type=int, help="device for training" |
| 40 | ) |
| 41 | parser.add_argument( |
| 42 | "--local_rank", default=0, type=int, help="local rank for dist training" |
| 43 | ) |
| 44 | parser.add_argument( |
| 45 | "--num_machines", default=1, type=int, help="num of node for training" |
| 46 | ) |
| 47 | parser.add_argument( |
| 48 | "--machine_rank", default=0, type=int, help="node rank for multi-node training" |
| 49 | ) |
| 50 | parser.add_argument( |
| 51 | "-f", |
| 52 | "--exp_file", |
| 53 | default=None, |
| 54 | type=str, |
| 55 | help="pls input your expriment description file", |
| 56 | ) |
| 57 | parser.add_argument( |
| 58 | "--fp16", |
| 59 | dest="fp16", |
| 60 | default=False, |
| 61 | action="store_true", |
| 62 | help="Adopting mix precision evaluating.", |
| 63 | ) |
| 64 | parser.add_argument( |
| 65 | "--fuse", |
| 66 | dest="fuse", |
| 67 | default=False, |
| 68 | action="store_true", |
| 69 | help="Fuse conv and bn for testing.", |
| 70 | ) |
| 71 | parser.add_argument( |
| 72 | "--trt", |
| 73 | dest="trt", |
| 74 | default=False, |
| 75 | action="store_true", |
| 76 | help="Using TensorRT model for testing.", |
| 77 | ) |
| 78 | parser.add_argument( |
| 79 | "--test", |