diff options
Diffstat (limited to 'nerv/tnn/layersT/lstm_t.lua')
| -rw-r--r-- | nerv/tnn/layersT/lstm_t.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/tnn/layersT/lstm_t.lua b/nerv/tnn/layersT/lstm_t.lua index ded6058..2a3342d 100644 --- a/nerv/tnn/layersT/lstm_t.lua +++ b/nerv/tnn/layersT/lstm_t.lua @@ -36,7 +36,7 @@ function LSTMLayerT:__init(id, global_conf, layer_conf) [ap("mainTanhL")] = {{}, {["dim_in"] = {self.dim_out[1]}, ["dim_out"] = {self.dim_out[1]}}}, [ap("outputTanhL")] = {{}, {["dim_in"] = {self.dim_out[1]}, ["dim_out"] = {self.dim_out[1]}}}, }, - ["nerv.GateFFFLayer"] = { + ["nerv.GateFLayer"] = { [ap("forgetGateL")] = {{}, {["dim_in"] = {self.dim_in[1], self.dim_in[2], self.dim_in[3]}, ["dim_out"] = {self.dim_in[3]}, ["pr"] = pr}}, [ap("inputGateL")] = {{}, {["dim_in"] = {self.dim_in[1], self.dim_in[2], self.dim_in[3]}, |