diff options
Diffstat (limited to 'nerv/layer/lstm_gate.lua')
| -rw-r--r-- | nerv/layer/lstm_gate.lua | 4 |
1 files changed, 2 insertions, 2 deletions
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 |