(model_path, net, prefix='rec')
| 40 | |
| 41 | |
| 42 | def load_model(model_path, net, prefix='rec'): |
| 43 | logger.info("start load model from {}".format(model_path)) |
| 44 | model_prefix = os.path.join(model_path, prefix) |
| 45 | param_state_dict = paddle.load(model_prefix + ".pdparams") |
| 46 | net.set_dict(param_state_dict) |
| 47 | |
| 48 | |
| 49 | def _mkdir_if_not_exist(path): |