save inference model to target path
(model_path,
epoch_id,
feed_vars,
fetch_vars,
exe,
prefix='rec_inference')
| 63 | |
| 64 | |
| 65 | def save_inference_model(model_path, |
| 66 | epoch_id, |
| 67 | feed_vars, |
| 68 | fetch_vars, |
| 69 | exe, |
| 70 | prefix='rec_inference'): |
| 71 | """ |
| 72 | save inference model to target path |
| 73 | """ |
| 74 | model_path = os.path.join(model_path, str(epoch_id)) |
| 75 | _mkdir_if_not_exist(model_path) |
| 76 | model_prefix = os.path.join(model_path, prefix) |
| 77 | paddle.static.save_inference_model( |
| 78 | path_prefix=model_prefix, |
| 79 | feed_vars=feed_vars, |
| 80 | fetch_vars=fetch_vars, |
| 81 | executor=exe) |
| 82 | |
| 83 | |
| 84 | def load_static_model(program, model_path, prefix='rec_static'): |
no test coverage detected