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

pull/2/head
Xinghai Sun 7 years ago
parent c8530dbbe8
commit 8f89a9bdd4

@ -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 \
--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(
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(

Loading…
Cancel
Save