File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed
Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 3737 help = "RNN layer number. (default: %(default)s)" )
3838parser .add_argument (
3939 "--rnn_layer_size" ,
40- default = 1280 ,
40+ default = 1024 ,
4141 type = int ,
4242 help = "RNN layer cell number. (default: %(default)s)" )
4343parser .add_argument (
4444 "--use_gru" ,
45- default = True ,
45+ default = False ,
4646 type = bool ,
4747 help = "Use GRU or simple RNN. (default: %(default)s)" )
4848parser .add_argument (
You can’t perform that action at this time.
0 commit comments