From cba54520413b4abb92d5c64d0b5c135dd9b9f1ef Mon Sep 17 00:00:00 2001 From: txh18 Date: Fri, 4 Dec 2015 21:55:20 +0800 Subject: small bug fix for lmptb.lstm_t_v2 --- nerv/examples/lmptb/lmptb/lstm_t_v2.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"] = { -- cgit v1.2.3