diff options
author | txh18 <cloudygooseg@gmail.com> | 2015-12-02 21:24:54 +0800 |
---|---|---|
committer | txh18 <cloudygooseg@gmail.com> | 2015-12-02 21:24:54 +0800 |
commit | df2a5d287c1889da0d3c91a2f057086b5a080be7 (patch) | |
tree | f3911221505e7cd06bbdc8bd25e119b25ef8e883 /nerv/examples/lmptb/m-tests/lmseqreader_test.lua | |
parent | 103a4291349c0f55155ca97bd236fc7784d286ff (diff) |
added se_mode for lmseqreader, todo:check it
Diffstat (limited to 'nerv/examples/lmptb/m-tests/lmseqreader_test.lua')
-rw-r--r-- | nerv/examples/lmptb/m-tests/lmseqreader_test.lua | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/nerv/examples/lmptb/m-tests/lmseqreader_test.lua b/nerv/examples/lmptb/m-tests/lmseqreader_test.lua index cbcdcbe..b98ff95 100644 --- a/nerv/examples/lmptb/m-tests/lmseqreader_test.lua +++ b/nerv/examples/lmptb/m-tests/lmseqreader_test.lua @@ -12,7 +12,7 @@ local batch_size = 3 local global_conf = { lrate = 1, wcost = 1e-6, momentum = 0, cumat_type = nerv.CuMatrixFloat, - mmat_type = nerv.CuMatrixFloat, + mmat_type = nerv.MMatrixFloat, hidden_size = 20, chunk_size = chunk_size, @@ -35,6 +35,7 @@ reader:open_file(test_fn) local feeds = {} feeds.flags_now = {} feeds.inputs_m = {} +feeds.flagsPack_now = {} for j = 1, chunk_size do feeds.inputs_m[j] = {global_conf.cumat_type(batch_size, 1), global_conf.cumat_type(batch_size, global_conf.vocab:size())} feeds.flags_now[j] = {} |