Main process for predict or export model
()
| 247 | |
| 248 | |
| 249 | def main(): |
| 250 | """Main process for predict or export model""" |
| 251 | opt = get_args(True) |
| 252 | set_parse(opt) |
| 253 | model_predict, config, rank = load_model(opt) |
| 254 | if opt.export: |
| 255 | export_mindir(model_predict, config) |
| 256 | else: |
| 257 | run_predict(model_predict, config, opt, rank) |
| 258 | |
| 259 | |
| 260 | if __name__ == "__main__": |
no test coverage detected