()
| 256 | |
| 257 | |
| 258 | def parse_args(): |
| 259 | parser = argparse.ArgumentParser() |
| 260 | parser.add_argument( |
| 261 | '--config', |
| 262 | type=str, |
| 263 | default="", |
| 264 | required=True, |
| 265 | help="the path of model yaml configuration file.") |
| 266 | parser.add_argument( |
| 267 | '--output', |
| 268 | type=str, |
| 269 | default="build", |
| 270 | help="output dir") |
| 271 | parser.add_argument( |
| 272 | "--enable_micro", |
| 273 | action="store_true", |
| 274 | help="enable convert micro.") |
| 275 | flgs, _ = parser.parse_known_args() |
| 276 | return flgs |
| 277 | |
| 278 | |
| 279 | if __name__ == '__main__': |