add depreciate and example changes

apache · Jul 25, 2018 · 0369ba3 · 0369ba3
1 parent 9a1b1a1
commit 0369ba3
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 7 deletions.
diff --git a/scala-package/core/src/main/scala/org/apache/mxnet/IO.scala b/scala-package/core/src/main/scala/org/apache/mxnet/IO.scala
@@ -354,9 +354,11 @@ abstract class DataIter extends Iterator[DataBatch] {
   def getIndex(): IndexedSeq[Long]
 
   // The name and shape of data provided by this iterator
+  @deprecated
   def provideData: ListMap[String, Shape]
 
   // The name and shape of label provided by this iterator
+  @deprecated
   def provideLabel: ListMap[String, Shape]
 
   // Provide type:DataDesc of the data

diff --git a/...ackage/examples/src/main/scala/org/apache/mxnetexamples/imclassification/TrainMnist.scala b/...ackage/examples/src/main/scala/org/apache/mxnetexamples/imclassification/TrainMnist.scala
@@ -77,7 +77,9 @@ object TrainMnist {
       "shuffle" -> "True",
       "flat" -> flat,
       "num_parts" -> kv.numWorkers.toString,
-      "part_index" -> kv.`rank`.toString))
+      "part_index" -> kv.`rank`.toString,
+      "dataLayout" -> "NT",
+      "labelLayout" -> "N"))
 
     val eval = IO.MNISTIter(Map(
       "image" -> (dataDir + "t10k-images-idx3-ubyte"),
@@ -87,7 +89,9 @@ object TrainMnist {
       "batch_size" -> batchSize.toString,
       "flat" -> flat,
       "num_parts" -> kv.numWorkers.toString,
-      "part_index" -> kv.`rank`.toString))
+      "part_index" -> kv.`rank`.toString,
+      "dataLayout" -> "NT",
+      "labelLayout" -> "N"))
 
     (train, eval)
   }

diff --git a/scala-package/examples/src/main/scala/org/apache/mxnetexamples/multitask/Data.scala b/scala-package/examples/src/main/scala/org/apache/mxnetexamples/multitask/Data.scala
@@ -21,9 +21,6 @@ import org.apache.mxnet.Shape
 import org.apache.mxnet.IO
 import org.apache.mxnet.DataIter
 
-/**
- * @author Depeng Liang
- */
 object Data {
 
   // return train and val iterators for mnist
@@ -35,15 +32,19 @@ object Data {
       "input_shape" -> inputShape.toString(),
       "batch_size" -> s"$batchSize",
       "shuffle" -> "True",
-      "flat" -> flat
+      "flat" -> flat,
+      "dataLayout" -> "NT",
+      "labelLayout" -> "N"
     )
     val trainDataIter = IO.MNISTIter(trainParams)
     val testParams = Map(
       "image" -> s"$dataPath/t10k-images-idx3-ubyte",
       "label" -> s"$dataPath/t10k-labels-idx1-ubyte",
       "input_shape" -> inputShape.toString(),
       "batch_size" -> s"$batchSize",
-      "flat" -> flat
+      "flat" -> flat,
+      "dataLayout" -> "NT",
+      "labelLayout" -> "N"
     )
     val testDataIter = IO.MNISTIter(testParams)
     (trainDataIter, testDataIter)