| 246 | |
| 247 | |
| 248 | def parse_args(): |
| 249 | parser = argparse.ArgumentParser() |
| 250 | parser.add_argument( |
| 251 | "--config", |
| 252 | type=str, |
| 253 | default="", |
| 254 | help="yaml conf path" |
| 255 | ) |
| 256 | parser.add_argument( |
| 257 | "--model_name", |
| 258 | type=str, |
| 259 | default="", |
| 260 | help="model name in yaml conf" |
| 261 | ) |
| 262 | parser.add_argument( |
| 263 | "--validate", |
| 264 | action="store_true", |
| 265 | help="enable validate" |
| 266 | ) |
| 267 | parser.add_argument( |
| 268 | "--validate_threshold", |
| 269 | type=float, |
| 270 | default="0.99", |
| 271 | help="validate threshold" |
| 272 | ) |
| 273 | parser.add_argument( |
| 274 | "--layers", |
| 275 | type=str, |
| 276 | default="-1", |
| 277 | help="'start_layer:end_layer' or 'layer', similar to python slice." |
| 278 | " Use with --validate flag.") |
| 279 | parser.add_argument( |
| 280 | "--backend", |
| 281 | type=str, |
| 282 | default="tensorflow", |
| 283 | help="onnx backend framework") |
| 284 | parser.add_argument( |
| 285 | "--build", |
| 286 | action="store_true", |
| 287 | help="if build before run" |
| 288 | ) |
| 289 | parser.add_argument( |
| 290 | '--output', |
| 291 | type=str, |
| 292 | default="build", |
| 293 | help="output dir") |
| 294 | parser.add_argument( |
| 295 | '--vlog_level', |
| 296 | type=int, |
| 297 | default="0", |
| 298 | help="vlog level") |
| 299 | parser.add_argument( |
| 300 | '--debug_mode', |
| 301 | action="store_true", |
| 302 | help="debug mode") |
| 303 | |
| 304 | return parser.parse_known_args() |
| 305 | |