diff options
| author | txh18 <[email protected]> | 2015-12-04 21:55:20 +0800 |
|---|---|---|
| committer | txh18 <[email protected]> | 2015-12-04 21:55:20 +0800 |
| commit | cba54520413b4abb92d5c64d0b5c135dd9b9f1ef (patch) | |
| tree | a8ecf92c1127ab22ac391fb760f207bba72ed372 | |
| parent | 0dbc40a998e2879f27110bd449f4cae9550c4e41 (diff) | |
small bug fix for lmptb.lstm_t_v2
| -rw-r--r-- | nerv/examples/lmptb/lmptb/lstm_t_v2.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/examples/lmptb/lmptb/lstm_t_v2.lua b/nerv/examples/lmptb/lmptb/lstm_t_v2.lua index d3fc920..dc2fe45 100644 --- a/nerv/examples/lmptb/lmptb/lstm_t_v2.lua +++ b/nerv/examples/lmptb/lmptb/lstm_t_v2.lua @@ -30,7 +30,7 @@ function LSTMLayerT:__init(id, global_conf, layer_conf) ["dim_out"] = {self.dim_in[3], self.dim_in[3]}, ["lambda"] = {1, 1}}}, }, ["nerv.AffineLayer"] = { - [ap("mainAffineL")] = {{}, {["dim_in"] = {self.dim_in[1], self.dim_in[2], self.dim_in[3]}, + [ap("mainAffineL")] = {{}, {["dim_in"] = {self.dim_in[1], self.dim_in[2]}, ["dim_out"] = {self.dim_out[1]}, ["pr"] = pr}}, }, ["nerv.TanhLayer"] = { |