diff options
Diffstat (limited to 'nerv/layer')
| -rw-r--r-- | nerv/layer/mse.lua | 2 |
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 |