diff options
| author | Determinant <ted.sybil@gmail.com> | 2016-03-02 18:24:09 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2016-03-02 18:24:09 +0800 |
| commit | ad704f2623cc9e0a5d702434bfdebc345465ca12 (patch) | |
| tree | 898d0688e913efc3ff098ba51e5c1a5488f5771d /nerv/examples/timit_baseline2.lua | |
| parent | d3abc6459a776ff7fa3777f4f561bc4f5d5e2075 (diff) | |
major changes in asr_trainer.lua; unified settings in `gconf`
Diffstat (limited to 'nerv/examples/timit_baseline2.lua')
| -rw-r--r-- | nerv/examples/timit_baseline2.lua | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/nerv/examples/timit_baseline2.lua b/nerv/examples/timit_baseline2.lua index 174b9e7..103d89d 100644 --- a/nerv/examples/timit_baseline2.lua +++ b/nerv/examples/timit_baseline2.lua @@ -1,8 +1,5 @@ require 'kaldi_io' -gconf = {lrate = 0.8, wcost = 1e-6, momentum = 0.9, - cumat_type = nerv.CuMatrixFloat, - mmat_type = nerv.MMatrixFloat, - frm_ext = 5, +gconf = {lrate = 0.8, wcost = 1e-6, momentum = 0.9, frm_ext = 5, tr_scp = "ark:/speechlab/tools/KALDI/kaldi-master/src/featbin/copy-feats " .. "scp:/speechlab/users/mfy43/timit/s5/exp/dnn4_nerv_prepare/train.scp ark:- |", cv_scp = "ark:/speechlab/tools/KALDI/kaldi-master/src/featbin/copy-feats " .. @@ -11,8 +8,7 @@ gconf = {lrate = 0.8, wcost = 1e-6, momentum = 0.9, "/speechlab/users/mfy43/timit/s5/exp/dnn4_nerv_prepare/nnet_output.nerv", "/speechlab/users/mfy43/timit/s5/exp/dnn4_nerv_prepare/nnet_trans.nerv"}, decode_param = {"/speechlab/users/mfy43/timit/nnet_init_20160229015745_iter_13_lr0.013437_tr72.434_cv58.729.nerv", - "/speechlab/users/mfy43/timit/s5/exp/dnn4_nerv_prepare/nnet_trans.nerv"}, - debug = false} + "/speechlab/users/mfy43/timit/s5/exp/dnn4_nerv_prepare/nnet_trans.nerv"}} function make_layer_repo(param_repo) local layer_repo = nerv.LayerRepo( @@ -183,6 +179,7 @@ function make_buffer(readers) return nerv.SGDBuffer(gconf, { buffer_size = gconf.buffer_size, + batch_size = gconf.batch_size, randomize = gconf.randomize, readers = readers, use_gpu = true |