diff options
| author | Determinant <ted.sybil@gmail.com> | 2016-03-16 17:53:39 +0800 |
|---|---|---|
| committer | Determinant <ted.sybil@gmail.com> | 2016-03-16 17:53:39 +0800 |
| commit | 289ac7f4b6e88b935da5c891e1efcf91fc047403 (patch) | |
| tree | d4fc3a4fc20f2d5908624b3f6587ecd57966d719 /nerv/examples/swb_baseline2.lua | |
| parent | 07fc1e2794027d44c255e1062c4491346b101a08 (diff) | |
merge seq_buffer and change asr_trainer.lua accordingly
Diffstat (limited to 'nerv/examples/swb_baseline2.lua')
| -rw-r--r-- | nerv/examples/swb_baseline2.lua | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/nerv/examples/swb_baseline2.lua b/nerv/examples/swb_baseline2.lua index 6796f6f..38cfb9a 100644 --- a/nerv/examples/swb_baseline2.lua +++ b/nerv/examples/swb_baseline2.lua @@ -6,7 +6,8 @@ gconf = {lrate = 0.8, wcost = 1e-6, momentum = 0.9, frm_ext = 5, cv_scp = "/speechlab/users/mfy43/swb50/train_cv.scp", htk_conf = "/speechlab/users/mfy43/swb50/plp_0_d_a.conf", initialized_param = {"/speechlab/users/mfy43/swb50/swb_init.nerv", - "/speechlab/users/mfy43/swb50/swb_global_transf.nerv"}} + "/speechlab/users/mfy43/swb50/swb_global_transf.nerv"}, + chunk_size = 1} function make_layer_repo(param_repo) local layer_repo = nerv.LayerRepo( @@ -145,7 +146,7 @@ end function make_readers(scp_file, layer_repo) return { - {reader = nerv.TNetReader(gconf, + {reader = nerv.HTKReader(gconf, { id = "main_scp", scp_file = scp_file, @@ -166,10 +167,11 @@ function make_readers(scp_file, layer_repo) end function make_buffer(readers) - return nerv.SGDBuffer(gconf, + return nerv.FrmBuffer(gconf, { buffer_size = gconf.buffer_size, batch_size = gconf.batch_size, + chunk_size = gconf.chunk_size, randomize = gconf.randomize, readers = readers, use_gpu = true |