(program, model_path, prefix='rec_static')
| 88 | |
| 89 | |
| 90 | def load_static_parameter(program, model_path, prefix='rec_static'): |
| 91 | logger.info("start load model from {}".format(model_path)) |
| 92 | model_prefix = os.path.join(model_path, prefix) |
| 93 | program_state = paddle.static.load_program_state(model_prefix) |
| 94 | paddle.static.set_program_state(program, program_state) |
| 95 | |
| 96 | |
| 97 | def save_data(fetch_batch_var, save_path, prefix='result'): |