diff options
| author | txh18 <cloudygooseg@gmail.com> | 2015-12-04 21:44:06 +0800 |
|---|---|---|
| committer | txh18 <cloudygooseg@gmail.com> | 2015-12-04 21:44:06 +0800 |
| commit | 0dbc40a998e2879f27110bd449f4cae9550c4e41 (patch) | |
| tree | 54e731db1304aaede37576bfee02b4b1145326fa /nerv/tnn | |
| parent | 7ee8988f21075246106a4d990190d0ef25fa82a8 (diff) | |
trying to use lstm_t_v2 for ptb
Diffstat (limited to 'nerv/tnn')
| -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]}, |