From bf335df65bd6943997d555fb4ec1ad2620494607 Mon Sep 17 00:00:00 2001 From: wangxiao Date: Tue, 3 Dec 2019 13:32:49 +0800 Subject: [PATCH] update download_models.py --- download_models.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/download_models.py b/download_models.py index 7095a91..e560d89 100644 --- a/download_models.py +++ b/download_models.py @@ -15,6 +15,10 @@ import paddlepalm as palm import sys + +items = ['ls', 'download'] +assert len(sys.argv) == 3, 'please input 2 parameters.' +assert sys.argv[1] in items, '{} is not found. Support methods: {}'.format(sys.argv[1], items) if(sys.argv[1] == 'ls'): palm.downloader.ls(sys.argv[2]) if(sys.argv[1] == 'download'): -- GitLab