fix download.py (PaddlePaddle#13)

an1018 · Nov 1, 2019 · a69ca0e · a69ca0e
1 parent 7ad69cc
commit a69ca0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ppdet/utils/download.py b/ppdet/utils/download.py
@@ -30,13 +30,13 @@
 import logging
 logger = logging.getLogger(__name__)
 
-__all__ = ['get_weights_path', 'get_dataset_path']
+__all__ = ['get_weights_path', 'get_dataset_path', 'download_dataset', 'create_voc_list']
 
 WEIGHTS_HOME = osp.expanduser("~/.cache/paddle/weights")
 DATASET_HOME = osp.expanduser("~/.cache/paddle/dataset")
 
 # dict of {dataset_name: (download_info, sub_dirs)}
-# download info: (url, md5sum)
+# download info: [(url, md5sum)]
 DATASETS = {
     'coco': ([
         (