| 126 | pass |
| 127 | |
| 128 | def add_args(self, parser) -> None: |
| 129 | general_group = parser.add_argument_group("general") |
| 130 | general_group.add_argument( |
| 131 | "--verbose", help="Display verbose logs.", action="store_true" |
| 132 | ) |
| 133 | general_group.add_argument( |
| 134 | "--info", help="Display info logs.", action="store_true" |
| 135 | ) |
| 136 | general_group.add_argument( |
| 137 | "--cleanup", |
| 138 | help="Cleans up user-specified workspace. Can not be cleaned if external files exist in workspace.", |
| 139 | action="store_false", |
| 140 | ) |
| 141 | general_group.add_argument( |
| 142 | "--working-dir", |
| 143 | help="Location of where to save the model and other downloaded files.", |
| 144 | required=True, |
| 145 | ) |
| 146 | general_group.add_argument( |
| 147 | "--batch-size", "-b", |
| 148 | help="Chosen batch size for given network", |
| 149 | required=False, |
| 150 | type=int, |
| 151 | default=1 |
| 152 | ) |
| 153 | |
| 154 | timing_group = parser.add_argument_group("inference measurement") |
| 155 | timing_group.add_argument( |
| 156 | "--iterations", |
| 157 | type=int, |
| 158 | help="Number of iterations to measure.", |
| 159 | default=self.DEFAULT_ITERATIONS, |
| 160 | ) |
| 161 | timing_group.add_argument( |
| 162 | "--number", |
| 163 | type=int, |
| 164 | help="Number of actual inference cycles per iterations.", |
| 165 | default=self.DEFAULT_NUMBER, |
| 166 | ) |
| 167 | timing_group.add_argument( |
| 168 | "--warmup", |
| 169 | type=int, |
| 170 | help="Number of warmup iterations before actual measurement occurs.", |
| 171 | default=self.DEFAULT_WARMUP, |
| 172 | ) |
| 173 | timing_group.add_argument( |
| 174 | "--duration", |
| 175 | type=float, |
| 176 | help="Minimal duration of inference iterations to measure.", |
| 177 | default=self.DEFAULT_DURATION, |
| 178 | ) |
| 179 | timing_group.add_argument( |
| 180 | "--percentile", |
| 181 | type=int, |
| 182 | help="Key percentile number for time measurement.", |
| 183 | default=self.DEFAULT_PERCENTILE, |
| 184 | ) |
| 185 | |