aboutsummaryrefslogtreecommitdiff
path: root/nerv/nn/trainer.lua
diff options
context:
space:
mode:
authorDeterminant <ted.sybil@gmail.com>2016-06-08 14:37:21 +0800
committerDeterminant <ted.sybil@gmail.com>2016-06-08 14:37:21 +0800
commit1d7af85c108d0e87b986db2dd23d524735a8b279 (patch)
tree24ba36d82f092390e133fce0eb0abe6647bb73e0 /nerv/nn/trainer.lua
parentb7cdd5da65a3e4ae58ffcfdf74710cfb1ee6327f (diff)
parentd58b7adf0acd68921ec2d38d5929bf68406d4982 (diff)
Merge remote-tracking branch 'lab/master'
Diffstat (limited to 'nerv/nn/trainer.lua')
-rw-r--r--nerv/nn/trainer.lua5
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