+ warning

ernie/file_utils.py

@@ -50,7 +50,6 @@ def _fetch_from_remote(url,
             cached_dir_model.mkdir()
             tmpfile = cached_dir_model / 'tmp'
             with tmpfile.open('wb') as f:
-                #url = 'https://ernie.bj.bcebos.com/ERNIE_stable.tgz'
                 r = requests.get(url, stream=True)
                 total_len = int(r.headers.get('content-length'))
                 for chunk in tqdm(

ernie/modeling_ernie.py

@@ -272,7 +272,7 @@ class PretrainedModel(object):
             pretrain_dir = Path(pretrain_dir_or_url)
 
         if not pretrain_dir.exists():
-            raise ValueError('pretrain dir not found: %s' % pretrain_dir)
+            raise ValueError('pretrain dir not found: %s, optional: %s' % (pretrain_dir, cls.resource_map.keys()))
         state_dict_path = pretrain_dir / 'saved_weights.pdparams'
         config_path = pretrain_dir / 'ernie_config.json'
 

ernie/tokenizing_ernie.py

@@ -107,7 +107,7 @@ class ErnieTokenizer(object):
                      (pretrain_dir_or_url, repr(cls.resource_map)))
             pretrain_dir = Path(pretrain_dir_or_url)
         if not pretrain_dir.exists():
-            raise ValueError('pretrain dir not found: %s' % pretrain_dir)
+            raise ValueError('pretrain dir not found: %s, optional: %s' % (pretrain_dir, cls.resource_map.keys()))
         vocab_path = pretrain_dir / 'vocab.txt'
         if not vocab_path.exists():
             raise ValueError('no vocab file in pretrain dir: %s' %