diff options
author | Determinant <ted.sybil@gmail.com> | 2016-06-08 14:37:21 +0800 |
---|---|---|
committer | Determinant <ted.sybil@gmail.com> | 2016-06-08 14:37:21 +0800 |
commit | 1d7af85c108d0e87b986db2dd23d524735a8b279 (patch) | |
tree | 24ba36d82f092390e133fce0eb0abe6647bb73e0 /nerv/nn/trainer.lua | |
parent | b7cdd5da65a3e4ae58ffcfdf74710cfb1ee6327f (diff) | |
parent | d58b7adf0acd68921ec2d38d5929bf68406d4982 (diff) |
Merge remote-tracking branch 'lab/master'
Diffstat (limited to 'nerv/nn/trainer.lua')
-rw-r--r-- | nerv/nn/trainer.lua | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/nerv/nn/trainer.lua b/nerv/nn/trainer.lua index 44390ea..a17b36c 100644 --- a/nerv/nn/trainer.lua +++ b/nerv/nn/trainer.lua @@ -25,7 +25,7 @@ function trainer:__init(gconf) self.input_order = self:get_input_order() self.network = nerv.Network('network', gconf, {network = graph, - clip = gconf.clip}) + nn_act_default = gconf.nn_act_default}) local network = self.network network:init(gconf.batch_size, gconf.chunk_size) @@ -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 |