Changed to set()0f without initializing a new array first

yzhliu · Dec 29, 2015 · acb82e2 · acb82e2
1 parent baf1279
commit acb82e2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/scala-package/core/src/main/scala/ml/dmlc/mxnet/Initializer.scala b/scala-package/core/src/main/scala/ml/dmlc/mxnet/Initializer.scala
@@ -56,19 +56,19 @@ abstract class Initializer {
   }
 
   def _initZero(name: String, arr: NDArray): Unit = {
-    arr.set(zeros(arr.size))
+    arr.set(0f)
   }
 
   def _initBias(name: String, arr: NDArray): Unit = {
-    arr.set(zeros(arr.size))
+    arr.set(0f)
   }
 
   def _initGamma(name: String, arr: NDArray): Unit = {
-    arr.set(ones(arr.size))
+    arr.set(1f)
   }
 
   def _initBeta(name: String, arr: NDArray): Unit = {
-    arr.set(zeros(arr.size))
+    arr.set(0f)
   }
 
   def _initWeight(name: String, arr: NDArray): Unit