From c73636ba680cdb5b57602a1876a75c110b43d426 Mon Sep 17 00:00:00 2001 From: Determinant Date: Mon, 18 Apr 2016 12:38:25 +0800 Subject: support SeqBuffer decoding; fix bugs in finding params --- nerv/layer/lstm_gate.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'nerv/layer/lstm_gate.lua') diff --git a/nerv/layer/lstm_gate.lua b/nerv/layer/lstm_gate.lua index e3b11b4..a3ae797 100644 --- a/nerv/layer/lstm_gate.lua +++ b/nerv/layer/lstm_gate.lua @@ -37,12 +37,12 @@ function LSTMGateLayer:init(batch_size) nerv.error("mismatching dimensions of linear transform parameter and output") end self.bp:train_init() - self.err_bakm = self.gconf.cumat_type(batch_size, self.dim_out[1]) + self.err_bakm = self.mat_type(batch_size, self.dim_out[1]) end function LSTMGateLayer:batch_resize(batch_size) if self.err_m:nrow() ~= batch_size then - self.err_bakm = self.gconf.cumat_type(batch_size, self.dim_out[1]) + self.err_bakm = self.mat_type(batch_size, self.dim_out[1]) end end -- cgit v1.2.3-70-g09d2