seq buffer support shuffle

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