aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authortxh18 <[email protected]>2015-12-04 21:55:20 +0800
committertxh18 <[email protected]>2015-12-04 21:55:20 +0800
commitcba54520413b4abb92d5c64d0b5c135dd9b9f1ef (patch)
treea8ecf92c1127ab22ac391fb760f207bba72ed372
parent0dbc40a998e2879f27110bd449f4cae9550c4e41 (diff)
small bug fix for lmptb.lstm_t_v2
-rw-r--r--nerv/examples/lmptb/lmptb/lstm_t_v2.lua2
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"] = {