aboutsummaryrefslogtreecommitdiff
path: root/nerv/layer/mse.lua
diff options
context:
space:
mode:
authorDeterminant <ted.sybil@gmail.com>2015-08-26 14:26:54 +0800
committerDeterminant <ted.sybil@gmail.com>2015-08-26 14:26:54 +0800
commite81e9832ec4f2ad031fd42b5018cea134e8cda7e (patch)
treeed49289619399a99c80f47398ccc4de9ae4cedf6 /nerv/layer/mse.lua
parented2a4148dbb9c18f428571b3e2970d7b2adfb058 (diff)
move global_transf to asr_trainer.lua
Diffstat (limited to 'nerv/layer/mse.lua')
-rw-r--r--nerv/layer/mse.lua2
1 files changed, 1 insertions, 1 deletions
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