init pretraining params
(exe,
pretraining_params_path,
main_program)
| 63 | |
| 64 | |
| 65 | def init_pretraining_params(exe, |
| 66 | pretraining_params_path, |
| 67 | main_program): |
| 68 | """init pretraining params""" |
| 69 | assert os.path.exists(pretraining_params_path |
| 70 | ), "[%s] cann't be found." % pretraining_params_path |
| 71 | |
| 72 | def existed_params(var): |
| 73 | if not isinstance(var, fluid.framework.Parameter): |
| 74 | return False |
| 75 | return os.path.exists(os.path.join(pretraining_params_path, var.name)) |
| 76 | |
| 77 | fluid.io.load_vars( |
| 78 | exe, |
| 79 | pretraining_params_path, |
| 80 | main_program=main_program, |
| 81 | predicate=existed_params) |
| 82 | print("Load pretraining parameters from {}.".format( |
| 83 | pretraining_params_path)) |
| 84 | |
| 85 | |
| 86 | def init_model(args, exe, startup_prog): |