fix bug for get_pretrain_weights

paddlex/__init__.py

@@ -25,4 +25,4 @@ load_model = cv.models.load_model
 datasets = cv.datasets
 
 log_level = 2
-__version__ = '0.1.0.github'
+__version__ = '0.1.1.github'

paddlex/cv/models/utils/pretrain_weights.py

@@ -76,10 +76,10 @@ def get_pretrain_weights(flag, model_type, backbone, save_dir):
         try:
             hub.download(backbone, save_path=new_save_dir)
         except Exception as e:
-            if isinstance(hub.ResourceNotFoundError):
+            if isinstance(e, hub.ResourceNotFoundError):
                 raise Exception(
                     "Resource for backbone {} not found".format(backbone))
-            elif isinstance(hub.ServerConnectionError):
+            elif isinstance(e, hub.ServerConnectionError):
                 raise Exception(
                     "Cannot get reource for backbone {}, please check your internet connecgtion"
                     .format(backbone))

setup.py

@@ -19,7 +19,7 @@ long_description = "PaddleX. A end-to-end deeplearning model development toolkit
 
 setuptools.setup(
     name="paddlex",
-    version='0.1.0',
+    version='0.1.1',
     author="paddlex",
     author_email="paddlex@baidu.com",
     description=long_description,