update data_iter example

kehao95 · Dec 13, 2015 · 8b81bb2 · 8b81bb2
1 parent f43d8f7
commit 8b81bb2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/example/python-howto/data_iter.py b/example/python-howto/data_iter.py
@@ -49,7 +49,11 @@
         round_batch=True)
 
 batchidx = 0
-for data, label in dataiter:
+for dbatch in dataiter:
+    data = dbatch.data[0]
+    label = dbatch.label[0]
+    pad = dbatch.pad
+    index = dbatch.index
     print("Batch", batchidx)
     print(label.asnumpy().flatten())
     batchidx += 1