From 4585970021f75d4c9e7154fc1681a80efa0f48ab Mon Sep 17 00:00:00 2001 From: Qi Liu Date: Mon, 9 May 2016 21:02:57 +0800 Subject: clip on gradient now --- nerv/nn/trainer.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'nerv/nn/trainer.lua') diff --git a/nerv/nn/trainer.lua b/nerv/nn/trainer.lua index 44390ea..8357c10 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) -- cgit v1.2.3 From e85a4af65640aef77378275d478c1ba8b06b785e Mon Sep 17 00:00:00 2001 From: Qi Liu Date: Wed, 11 May 2016 16:53:56 +0800 Subject: seq buffer support shuffle --- nerv/nn/trainer.lua | 3 +++ 1 file changed, 3 insertions(+) (limited to 'nerv/nn/trainer.lua') 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 -- cgit v1.2.3