From 8f89a9bdd49a77b49c756700440a19bcd57a6667 Mon Sep 17 00:00:00 2001
From: Xinghai Sun <sunxinghai1216@gmail.com>
Date: Mon, 4 Sep 2017 13:06:54 +0800
Subject: [PATCH] Print log to pfs for DS cloud training and set use_gru to
 False by default.

---
 cloud/pcloud_train.sh | 4 ++--
 train.py              | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/cloud/pcloud_train.sh b/cloud/pcloud_train.sh
index e42da1d6..75949574 100644
--- a/cloud/pcloud_train.sh
+++ b/cloud/pcloud_train.sh
@@ -13,7 +13,7 @@ python ./cloud/split_data.py \
 --in_manifest_path=${DEV_MANIFEST} \
 --out_manifest_path='/local.manifest.dev'
 
-python train.py \
+python -u train.py \
 --batch_size=$BATCH_SIZE \
 --use_gpu=1 \
 --trainer_count=${NUM_GPU} \
@@ -21,4 +21,4 @@ python train.py \
 --is_local=${IS_LOCAL} \
 --train_manifest_path='/local.manifest.train' \
 --dev_manifest_path='/local.manifest.dev' \
---output_model_dir=${MODEL_PATH} \
+--output_model_dir=${MODEL_PATH} 2>&1 | tee ./log/train.log
diff --git a/train.py b/train.py
index 8e95d7bc..1d0b92ff 100644
--- a/train.py
+++ b/train.py
@@ -37,12 +37,12 @@ parser.add_argument(
     help="RNN layer number. (default: %(default)s)")
 parser.add_argument(
     "--rnn_layer_size",
-    default=1280,
+    default=1024,
     type=int,
     help="RNN layer cell number. (default: %(default)s)")
 parser.add_argument(
     "--use_gru",
-    default=True,
+    default=False,
     type=bool,
     help="Use GRU or simple RNN. (default: %(default)s)")
 parser.add_argument(
-- 
GitLab