Procházet zdrojové kódy

[Feat] Add warning when raw_params not exist

Bobholamovic před 2 roky
rodič
revize
5c36d25623
1 změnil soubory, kde provedl 4 přidání a 0 odebrání
  1. 4 0
      paddlers/tasks/load_model.py

+ 4 - 0
paddlers/tasks/load_model.py

@@ -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)