diff options
Diffstat (limited to 'nerv/layer/gru.lua')
| -rw-r--r-- | nerv/layer/gru.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/layer/gru.lua b/nerv/layer/gru.lua index 2162e28..e81d21a 100644 --- a/nerv/layer/gru.lua +++ b/nerv/layer/gru.lua @@ -48,7 +48,7 @@ function GRULayer:__init(id, global_conf, layer_conf) ["nerv.TanhLayer"] = { [ap("mainTanhL")] = {{}, {dim_in = {dout1}, dim_out = {dout1}}}, }, - ["nerv.GateFLayer"] = { + ["nerv.LSTMGateLayer"] = { [ap("resetGateL")] = {{}, {dim_in = {din1, din2}, dim_out = {din2}, pr = pr}}, |