提交 8f89a9bd 编写于 作者: X Xinghai Sun

Print log to pfs for DS cloud training and set use_gru to False by default.

上级 c8530dbb
......@@ -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
......@@ -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(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册