From eccb3ab6b0f86b8d773ee5fdc153c6faa65cdda9 Mon Sep 17 00:00:00 2001
From: wangxiao <zko1021@foxmail.com>
Date: Fri, 29 Nov 2019 14:37:49 +0800
Subject: [PATCH] update download_models.py & README.md

---
 README.md          | 2 +-
 download_models.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 700a7eb..453453e 100644
--- a/README.md
+++ b/README.md
@@ -113,7 +113,7 @@ paddlepalm框架的运行原理图如图所示
 我们提供了BERT、ERNIE等主干网络的相关预训练模型。为了加速模型收敛，获得更佳的测试集表现，我们强烈建议用户在多任务学习时尽量在预训练模型的基础上进行（而不是从参数随机初始化开始）。用户可以查看可供下载的预训练模型：
 
 ```shell
-python download_models.py
+python download_models.py ls pretrain
 ```
 
 用户可通过运行`python download_models.py download <model_name>`下载需要的预训练模型，例如，下载预训练BERT模型（uncased large）的命令如下：
diff --git a/download_models.py b/download_models.py
index 620e84d..4c7314e 100644
--- a/download_models.py
+++ b/download_models.py
@@ -16,6 +16,6 @@
 import paddlepalm as palm
 import sys
 if(sys.argv[1] == 'ls'):
-  palm.ls(sys.argv[1], sys.argv[2])
+  palm.ls(sys.argv[2])
 if(sys.argv[1] == 'download'):
-  palm.download(sys.argv[1], sys.argv[2])
+  palm.download('pretrain', sys.argv[2])
-- 
GitLab