Main process for predict or export model
()
| 296 | |
| 297 | |
| 298 | def main(): |
| 299 | """Main process for predict or export model""" |
| 300 | print("===Enter main!") |
| 301 | opt = get_args(True) |
| 302 | set_parse(opt) |
| 303 | model_predict, config, rank = load_model(opt) |
| 304 | if opt.export: |
| 305 | export_mindir(model_predict, config) |
| 306 | else: |
| 307 | run_predict(model_predict, config, opt, rank) |
| 308 | |
| 309 | |
| 310 | if __name__ == "__main__": |
no test coverage detected