diff options
| -rw-r--r-- | nerv/examples/seq_trainer.lua | 1 |
1 files changed, 1 insertions, 0 deletions
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 |