Skip to content

Commit b0d5050

Browse files
authored
Merge pull request #221 from xinghai-sun/network_structure
Print log to pfs for cloud training and set use_gru to False by default.
2 parents 0d44c20 + b6b180a commit b0d5050

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

deep_speech_2/cloud/pcloud_train.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,12 +13,12 @@ python ./cloud/split_data.py \
1313
--in_manifest_path=${DEV_MANIFEST} \
1414
--out_manifest_path='/local.manifest.dev'
1515

16-
python train.py \
16+
python -u train.py \
1717
--batch_size=$BATCH_SIZE \
1818
--use_gpu=1 \
1919
--trainer_count=${NUM_GPU} \
2020
--num_threads_data=${NUM_GPU} \
2121
--is_local=${IS_LOCAL} \
2222
--train_manifest_path='/local.manifest.train' \
2323
--dev_manifest_path='/local.manifest.dev' \
24-
--output_model_dir=${MODEL_PATH} \
24+
--output_model_dir=${MODEL_PATH} 2>&1 | tee ./log/train.log

deep_speech_2/train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,12 @@
3737
help="RNN layer number. (default: %(default)s)")
3838
parser.add_argument(
3939
"--rnn_layer_size",
40-
default=1280,
40+
default=1024,
4141
type=int,
4242
help="RNN layer cell number. (default: %(default)s)")
4343
parser.add_argument(
4444
"--use_gru",
45-
default=True,
45+
default=False,
4646
type=bool,
4747
help="Use GRU or simple RNN. (default: %(default)s)")
4848
parser.add_argument(

0 commit comments

Comments
 (0)