Skip to content

Commit

Permalink
[scripts] tfrnnlm: Modify TensorFlow flag format for compatibility wi…
Browse files Browse the repository at this point in the history
…th recent versions (#2760)
  • Loading branch information
ervaibhavkumar authored and danpovey committed Oct 4, 2018
1 parent 8b8604c commit 2d163f4
Show file tree
Hide file tree
Showing 6 changed files with 24 additions and 24 deletions.
2 changes: 1 addition & 1 deletion egs/ami/s5/local/tfrnnlm/run_lstm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ mkdir -p $dir
if [ $stage -le 2 ]; then
# the following script uses TensorFlow. You could use tools/extras/install_tensorflow_py.sh to install it
$cuda_cmd $dir/train_rnnlm.log utils/parallel/limit_num_gpus.sh \
python steps/tfrnnlm/lstm.py --data-path=$dir --save-path=$dir/rnnlm --vocab-path=$dir/wordlist.rnn.final
python steps/tfrnnlm/lstm.py --data_path=$dir --save_path=$dir/rnnlm --vocab_path=$dir/wordlist.rnn.final
fi

final_lm=ami_fsh.o3g.kn
Expand Down
2 changes: 1 addition & 1 deletion egs/ami/s5/local/tfrnnlm/run_lstm_fast.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ mkdir -p $dir
if [ $stage -le 2 ]; then
# the following script uses TensorFlow. You could use tools/extras/install_tensorflow_py.sh to install it
$cuda_cmd $dir/train_rnnlm.log utils/parallel/limit_num_gpus.sh \
python steps/tfrnnlm/lstm_fast.py --data-path=$dir --save-path=$dir/rnnlm --vocab-path=$dir/wordlist.rnn.final
python steps/tfrnnlm/lstm_fast.py --data_path=$dir --save_path=$dir/rnnlm --vocab_path=$dir/wordlist.rnn.final
fi

final_lm=ami_fsh.o3g.kn
Expand Down
2 changes: 1 addition & 1 deletion egs/ami/s5/local/tfrnnlm/run_vanilla_rnnlm.sh
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ mkdir -p $dir
if [ $stage -le 2 ]; then
# the following script uses TensorFlow. You could use tools/extras/install_tensorflow_py.sh to install it
$cuda_cmd $dir/train_rnnlm.log utils/parallel/limit_num_gpus.sh \
python steps/tfrnnlm/vanilla_rnnlm.py --data-path=$dir --save-path=$dir/rnnlm --vocab-path=$dir/wordlist.rnn.final
python steps/tfrnnlm/vanilla_rnnlm.py --data_path=$dir --save_path=$dir/rnnlm --vocab_path=$dir/wordlist.rnn.final
fi

final_lm=ami_fsh.o3g.kn
Expand Down
14 changes: 7 additions & 7 deletions egs/wsj/s5/steps/tfrnnlm/lstm.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,8 @@

# this script trains a vanilla RNNLM with TensorFlow.
# to call the script, do
# python steps/tfrnnlm/lstm.py --data-path=$datadir \
# --save-path=$savepath --vocab-path=$rnn.wordlist [--hidden-size=$size]
# python steps/tfrnnlm/lstm.py --data_path=$datadir \
# --save_path=$savepath --vocab_path=$rnn.wordlist [--hidden-size=$size]
#
# One example recipe is at egs/ami/s5/local/tfrnnlm/run_lstm.sh

Expand All @@ -38,15 +38,15 @@
flags = tf.flags
logging = tf.logging

flags.DEFINE_integer("hidden-size", 200, "hidden dim of RNN")
flags.DEFINE_integer("hidden_size", 200, "hidden dim of RNN")

flags.DEFINE_string("data-path", None,
flags.DEFINE_string("data_path", None,
"Where the training/test data is stored.")
flags.DEFINE_string("vocab-path", None,
flags.DEFINE_string("vocab_path", None,
"Where the wordlist file is stored.")
flags.DEFINE_string("save-path", None,
flags.DEFINE_string("save_path", None,
"Model output directory.")
flags.DEFINE_bool("use-fp16", False,
flags.DEFINE_bool("use_fp16", False,
"Train using 16-bit floats instead of 32bit floats")

FLAGS = flags.FLAGS
Expand Down
14 changes: 7 additions & 7 deletions egs/wsj/s5/steps/tfrnnlm/lstm_fast.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,8 @@

# this script trains a vanilla RNNLM with TensorFlow.
# to call the script, do
# python steps/tfrnnlm/lstm_fast.py --data-path=$datadir \
# --save-path=$savepath --vocab-path=$rnn.wordlist [--hidden-size=$size]
# python steps/tfrnnlm/lstm_fast.py --data_path=$datadir \
# --save_path=$savepath --vocab_path=$rnn.wordlist [--hidden-size=$size]
#
# One example recipe is at egs/ami/s5/local/tfrnnlm/run_vanilla_rnnlm.sh

Expand All @@ -38,15 +38,15 @@
flags = tf.flags
logging = tf.logging

flags.DEFINE_integer("hidden-size", 200, "hidden dim of RNN")
flags.DEFINE_integer("hidden_size", 200, "hidden dim of RNN")

flags.DEFINE_string("data-path", None,
flags.DEFINE_string("data_path", None,
"Where the training/test data is stored.")
flags.DEFINE_string("vocab-path", None,
flags.DEFINE_string("vocab_path", None,
"Where the wordlist file is stored.")
flags.DEFINE_string("save-path", None,
flags.DEFINE_string("save_path", None,
"Model output directory.")
flags.DEFINE_bool("use-fp16", False,
flags.DEFINE_bool("use_fp16", False,
"Train using 16-bit floats instead of 32bit floats")

FLAGS = flags.FLAGS
Expand Down
14 changes: 7 additions & 7 deletions egs/wsj/s5/steps/tfrnnlm/vanilla_rnnlm.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,8 +16,8 @@

# this script trains a vanilla RNNLM with TensorFlow.
# to call the script, do
# python steps/tfrnnlm/vanilla_rnnlm.py --data-path=$datadir \
# --save-path=$savepath --vocab-path=$rnn.wordlist [--hidden-size=$size]
# python steps/tfrnnlm/vanilla_rnnlm.py --data_path=$datadir \
# --save_path=$savepath --vocab_path=$rnn.wordlist [--hidden-size=$size]
#
# One example recipe is at egs/ami/s5/local/tfrnnlm/run_vanilla_rnnlm.sh

Expand All @@ -38,15 +38,15 @@
flags = tf.flags
logging = tf.logging

flags.DEFINE_integer("hidden-size", 200, "hidden dim of RNN")
flags.DEFINE_integer("hidden_size", 200, "hidden dim of RNN")

flags.DEFINE_string("data-path", None,
flags.DEFINE_string("data_path", None,
"Where the training/test data is stored.")
flags.DEFINE_string("vocab-path", None,
flags.DEFINE_string("vocab_path", None,
"Where the wordlist file is stored.")
flags.DEFINE_string("save-path", None,
flags.DEFINE_string("save_path", None,
"Model output directory.")
flags.DEFINE_bool("use-fp16", False,
flags.DEFINE_bool("use_fp16", False,
"Train using 16-bit floats instead of 32bit floats")

FLAGS = flags.FLAGS
Expand Down

0 comments on commit 2d163f4

Please sign in to comment.