diff options
| -rw-r--r-- | nerv/layer/lstm.lua | 12 |
1 files 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"] = { |