From b150ce2e3b213a06ea1882c39cfa59c51690e525 Mon Sep 17 00:00:00 2001
From: guosheng <guosheng@baidu.com>
Date: Mon, 13 Aug 2018 11:56:57 +0800
Subject: [PATCH] Fix ce data in Transformer

---
 fluid/neural_machine_translation/transformer/.run_ce.sh | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/fluid/neural_machine_translation/transformer/.run_ce.sh b/fluid/neural_machine_translation/transformer/.run_ce.sh
index 50161ca4..492150b3 100644
--- a/fluid/neural_machine_translation/transformer/.run_ce.sh
+++ b/fluid/neural_machine_translation/transformer/.run_ce.sh
@@ -1,9 +1,8 @@
 #!/bin/bash
 
 DATA_PATH=$HOME/.cache/paddle/dataset/wmt16
-if [ ! -d $DATA_PATH ] ; then
-    python -c 'import paddle;paddle.dataset.wmt16.train(10000, 10000, "en")'\
-        '().next()'
+if [ ! -d $DATA_PATH/en_10000.dict ] ; then
+    python -c 'import paddle;paddle.dataset.wmt16.train(10000, 10000, "en")().next()'
     tar -zxf $DATA_PATH/wmt16.tar.gz -C $DATA_PATH
 fi
 
-- 
GitLab