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