aboutsummaryrefslogtreecommitdiff
path: root/nerv/layer
diff options
context:
space:
mode:
Diffstat (limited to 'nerv/layer')
-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"] = {