From 60e7718f4db981557ac48d9d375d8e23b1cd39d1 Mon Sep 17 00:00:00 2001 From: Yimmon Zhuang Date: Sat, 10 Oct 2015 19:06:53 +0800 Subject: disable batch when sequence training --- nerv/examples/seq_trainer.lua | 1 + 1 file changed, 1 insertion(+) diff --git a/nerv/examples/seq_trainer.lua b/nerv/examples/seq_trainer.lua index df96e68..b8ed3eb 100644 --- a/nerv/examples/seq_trainer.lua +++ b/nerv/examples/seq_trainer.lua @@ -49,6 +49,7 @@ function build_trainer(ifname) network:batch_resize(input[1]:nrow()) if network:propagate(input, {{}}) == true then network:back_propagate({{}}, err_output, input, {{}}) + gconf.batch_size = 1.0 - gconf.momentum -- important!!! network:update({{}}, input, {{}}) end -- collect garbage in-time to save GPU memory -- cgit v1.2.3