aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorQi Liu <[email protected]>2016-02-24 16:40:54 +0800
committerQi Liu <[email protected]>2016-02-24 16:40:54 +0800
commit14c1997203e04838b1737716dc385e1aa08fe91f (patch)
tree549d7abfa0a2df3b7af14cf44d41040e898db3d6
parent9fb40644f5b7aa141ca23c93e6aa77c05a04c76c (diff)
fix bug
-rw-r--r--nerv/layer/lstm.lua12
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"] = {