提交 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 \ ...@@ -13,7 +13,7 @@ python ./cloud/split_data.py \
--in_manifest_path=${DEV_MANIFEST} \ --in_manifest_path=${DEV_MANIFEST} \
--out_manifest_path='/local.manifest.dev' --out_manifest_path='/local.manifest.dev'
python train.py \ python -u train.py \
--batch_size=$BATCH_SIZE \ --batch_size=$BATCH_SIZE \
--use_gpu=1 \ --use_gpu=1 \
--trainer_count=${NUM_GPU} \ --trainer_count=${NUM_GPU} \
...@@ -21,4 +21,4 @@ python train.py \ ...@@ -21,4 +21,4 @@ python train.py \
--is_local=${IS_LOCAL} \ --is_local=${IS_LOCAL} \
--train_manifest_path='/local.manifest.train' \ --train_manifest_path='/local.manifest.train' \
--dev_manifest_path='/local.manifest.dev' \ --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( ...@@ -37,12 +37,12 @@ parser.add_argument(
help="RNN layer number. (default: %(default)s)") help="RNN layer number. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--rnn_layer_size", "--rnn_layer_size",
default=1280, default=1024,
type=int, type=int,
help="RNN layer cell number. (default: %(default)s)") help="RNN layer cell number. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gru", "--use_gru",
default=True, default=False,
type=bool, type=bool,
help="Use GRU or simple RNN. (default: %(default)s)") help="Use GRU or simple RNN. (default: %(default)s)")
parser.add_argument( parser.add_argument(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册