diff options
author | Determinant <ted.sybil@gmail.com> | 2016-03-16 17:56:40 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-03-16 17:56:40 +0800 |
commit | 00211587988024e34370e7ec2ee1cb1c8632f116 (patch) | |
tree | d4fc3a4fc20f2d5908624b3f6587ecd57966d719 /nerv/examples/swb_baseline2.lua | |
parent | a5a4d2735b595fc9fadc9c7e91198786d3c0e078 (diff) | |
parent | 289ac7f4b6e88b935da5c891e1efcf91fc047403 (diff) |
Merge branch 'liuq901-master' into seq-buffer
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 |