|
|
@ -82,6 +82,10 @@ def load_model(model_dir, **params): |
|
|
|
model_info['_init_params'].update({'with_net': with_net}) |
|
|
|
model_info['_init_params'].update({'with_net': with_net}) |
|
|
|
|
|
|
|
|
|
|
|
with paddle.utils.unique_name.guard(): |
|
|
|
with paddle.utils.unique_name.guard(): |
|
|
|
|
|
|
|
if 'raw_params' not in model_info: |
|
|
|
|
|
|
|
logging.warning( |
|
|
|
|
|
|
|
"Cannot find raw_params. Default arguments will be used to construct the model." |
|
|
|
|
|
|
|
) |
|
|
|
params = model_info.pop('raw_params', {}) |
|
|
|
params = model_info.pop('raw_params', {}) |
|
|
|
params.update(model_info['_init_params']) |
|
|
|
params.update(model_info['_init_params']) |
|
|
|
model = getattr(paddlers.tasks, model_info['Model'])(**params) |
|
|
|
model = getattr(paddlers.tasks, model_info['Model'])(**params) |
|
|
|