From 5c36d2562320243269bcfb189614761473ea0fa4 Mon Sep 17 00:00:00 2001 From: Bobholamovic Date: Tue, 24 May 2022 12:37:22 +0800 Subject: [PATCH] [Feat] Add warning when raw_params not exist --- paddlers/tasks/load_model.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/paddlers/tasks/load_model.py b/paddlers/tasks/load_model.py index fb82a7f..9d05be3 100644 --- a/paddlers/tasks/load_model.py +++ b/paddlers/tasks/load_model.py @@ -82,6 +82,10 @@ def load_model(model_dir, **params): model_info['_init_params'].update({'with_net': with_net}) 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.update(model_info['_init_params']) model = getattr(paddlers.tasks, model_info['Model'])(**params)