From e81e9832ec4f2ad031fd42b5018cea134e8cda7e Mon Sep 17 00:00:00 2001 From: Determinant Date: Wed, 26 Aug 2015 14:26:54 +0800 Subject: move global_transf to asr_trainer.lua --- nerv/layer/mse.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'nerv/layer/mse.lua') diff --git a/nerv/layer/mse.lua b/nerv/layer/mse.lua index 9a97add..2516998 100644 --- a/nerv/layer/mse.lua +++ b/nerv/layer/mse.lua @@ -34,7 +34,7 @@ function MSELayer:propagate(input, output) if output[1] ~= nil then output[1]:copy_fromd(mse_sum) end - self.total_mse = self.total_mse + mse_sum:colsum()[0] + self.total_mse = self.total_mse + mse_sum:colsum()[0][0] self.total_frames = self.total_frames + mse_sum:nrow() end -- cgit v1.2.3