From c0f8360e47d7dfac4ddd633625b12e0382aeaff7 Mon Sep 17 00:00:00 2001
From: mmglove <38800877+mmglove@users.noreply.github.com>
Date: Mon, 6 Jan 2020 19:10:34 +0800
Subject: [PATCH] Add word2vec ce (#4163)

* add compress ce

* add ce for word2vec
---
 PaddleRec/word2vec/train.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/PaddleRec/word2vec/train.py b/PaddleRec/word2vec/train.py
index 087c9b1d..929abf4d 100644
--- a/PaddleRec/word2vec/train.py
+++ b/PaddleRec/word2vec/train.py
@@ -84,6 +84,9 @@ def parse_args():
         required=False,
         default=False,
         help='print speed or not , (default: False)')
+    parser.add_argument(
+        '--enable_ce', action='store_true', help='If set, run the task with continuous evaluation logs.')
+
     return parser.parse_args()
 
 
@@ -195,6 +198,11 @@ def GetFileList(data_path):
 
 
 def train(args):
+    # add ce
+    if args.enable_ce:
+        SEED = 102
+        fluid.default_main_program().random_seed = SEED
+        fluid.default_startup_program().random_seed = SEED
 
     if not os.path.isdir(args.model_output_dir):
         os.mkdir(args.model_output_dir)
-- 
GitLab