Main process for predict or export model
()
| 206 | |
| 207 | |
| 208 | def main(): |
| 209 | """Main process for predict or export model""" |
| 210 | print("===Enter main!") |
| 211 | opt = get_args(True) |
| 212 | set_parse(opt) |
| 213 | model_predict, config, rank = load_model(opt) |
| 214 | run_predict(model_predict, config, opt, rank) |
| 215 | |
| 216 | |
| 217 | if __name__ == "__main__": |
no test coverage detected