Main process for predict or export model
()
| 241 | |
| 242 | |
| 243 | def main(): |
| 244 | """Main process for predict or export model""" |
| 245 | opt = get_args(True) |
| 246 | set_parse(opt) |
| 247 | model_predict, config, rank = load_model(opt) |
| 248 | if opt.export: |
| 249 | export_mindir(model_predict, config) |
| 250 | else: |
| 251 | run_predict(model_predict, config, opt, rank) |
| 252 | |
| 253 | |
| 254 | if __name__ == "__main__": |
no test coverage detected