From 14c1997203e04838b1737716dc385e1aa08fe91f Mon Sep 17 00:00:00 2001 From: Qi Liu Date: Wed, 24 Feb 2016 16:40:54 +0800 Subject: fix bug --- nerv/layer/lstm.lua | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/nerv/layer/lstm.lua b/nerv/layer/lstm.lua index a1833ef..d8eee71 100644 --- a/nerv/layer/lstm.lua +++ b/nerv/layer/lstm.lua @@ -49,14 +49,14 @@ function LSTMLayer:__init(id, global_conf, layer_conf) }, ["nerv.LSTMGateLayer"] = { [ap("forgetGateL")] = {{}, {dim_in = {din1, din2, din3}, - dim_out = {din3}, pr = pr}, - param_type = {'N', 'N', 'D'}}, + dim_out = {din3}, pr = pr, + param_type = {'N', 'N', 'D'}}}, [ap("inputGateL")] = {{}, {dim_in = {din1, din2, din3}, - dim_out = {din3}, pr = pr}, - param_type = {'N', 'N', 'D'}}, + dim_out = {din3}, pr = pr, + param_type = {'N', 'N', 'D'}}}, [ap("outputGateL")] = {{}, {dim_in = {din1, din2, din3}, - dim_out = {din3}, pr = pr}, - param_type = {'N', 'N', 'D'}}, + dim_out = {din3}, pr = pr, + param_type = {'N', 'N', 'D'}}}, }, ["nerv.ElemMulLayer"] = { -- cgit v1.2.3