Update julia/src/metric.jl

Co-Authored-By: Iblis Lin <[email protected]>
apache · Oct 27, 2019 · 2dfa1cc · 2dfa1cc
1 parent c72507b
commit 2dfa1cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/julia/src/metric.jl b/julia/src/metric.jl
@@ -260,7 +260,7 @@ end
 
 function get(metric::MSE)
   # Delay copy until last possible moment
-  mse_sum = mapreduce(nda->copy(nda)[1], +, metric.mse_sum)
+  mse_sum = mapreduce(nda->copy(nda)[1], +, metric.mse_sum; init = zero(MX_float))
   [(:MSE, mse_sum / metric.n_sample)]
 end