From 19e0d603445227133d1aafef673e27be097f046d Mon Sep 17 00:00:00 2001
From: tangwei12 <tangwei12@baidu.com>
Date: Wed, 21 Nov 2018 13:53:43 +0800
Subject: [PATCH] add is_sparse and args optimize

---
 fluid/PaddleRec/word2vec/train.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/fluid/PaddleRec/word2vec/train.py b/fluid/PaddleRec/word2vec/train.py
index 55741116..6bac5e6a 100644
--- a/fluid/PaddleRec/word2vec/train.py
+++ b/fluid/PaddleRec/word2vec/train.py
@@ -78,7 +78,8 @@ def parse_args():
         help='max code length used by hierarchical sigmoid, (default: 40)')
     parser.add_argument(
         '--is_sparse',
-        type=bool,
+        action='store_true',
+        required=False,
         default=False,
         help='embedding and nce will use sparse or not, (default: False)')
 
-- 
GitLab