diff options
author | Qi Liu <liuq901@163.com> | 2016-05-11 16:53:56 +0800 |
---|---|---|
committer | Qi Liu <liuq901@163.com> | 2016-05-11 16:53:56 +0800 |
commit | e85a4af65640aef77378275d478c1ba8b06b785e (patch) | |
tree | d7c95e62c067ec510fd3ad8b546fc367805ab632 /nerv/nn/trainer.lua | |
parent | 4585970021f75d4c9e7154fc1681a80efa0f48ab (diff) |
seq buffer support shuffle
Diffstat (limited to 'nerv/nn/trainer.lua')
-rw-r--r-- | nerv/nn/trainer.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/nerv/nn/trainer.lua b/nerv/nn/trainer.lua index 8357c10..a17b36c 100644 --- a/nerv/nn/trainer.lua +++ b/nerv/nn/trainer.lua @@ -77,9 +77,12 @@ function trainer:make_buffer(readers) }) else return nerv.SeqBuffer(gconf, { + buffer_size = gconf.buffer_size, batch_size = gconf.batch_size, chunk_size = gconf.chunk_size, + randomize = gconf.randomize, readers = readers, + nn_act_default = gconf.nn_act_default, }) end end |