Fix Wrong labelSize at Sample.scala 411 (#2147) (#2149)

"val labelSize = features.map(_.size())" -> "val labelSize = labels.map(_.size())"
intel-analytics · Jan 3, 2018 · eb5d1a1 · eb5d1a1
1 parent 4d1058a
commit eb5d1a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spark/dl/src/main/scala/com/intel/analytics/bigdl/dataset/Sample.scala b/spark/dl/src/main/scala/com/intel/analytics/bigdl/dataset/Sample.scala
@@ -408,7 +408,7 @@ private[bigdl] class TensorSample[T: ClassTag](
       val features: Array[Tensor[T]],
       val labels: Array[Tensor[T]]) extends Sample[T] {
   val featureSize = features.map(_.size())
-  val labelSize = features.map(_.size())
+  val labelSize = labels.map(_.size())
 
   def featureLength(index: Int): Int = {
     features(0).size(1)