diff --git a/example/python-howto/data_iter.py b/example/python-howto/data_iter.py index ea541b6985ef..98677cbd4cc2 100644 --- 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