diff options
| author | Qi Liu <liuq901@163.com> | 2016-02-24 16:40:54 +0800 |
|---|---|---|
| committer | Qi Liu <liuq901@163.com> | 2016-02-24 16:40:54 +0800 |
| commit | 14c1997203e04838b1737716dc385e1aa08fe91f (patch) | |
| tree | 549d7abfa0a2df3b7af14cf44d41040e898db3d6 | |
| parent | 9fb40644f5b7aa141ca23c93e6aa77c05a04c76c (diff) | |
fix bug
| -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"] = { |