Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Add amend
Browse files Browse the repository at this point in the history
  • Loading branch information
lanking520 committed Jul 23, 2018
1 parent 4a46359 commit 9332064
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,8 @@ private[mxnet] class MXDataIter(private[mxnet] val handle: DataIterHandle,
checkCall(_LIB.mxDataIterNext(handle, next))
if (next.value > 0) {
currentBatch = new DataBatch(data = getData(), label = getLabel(),
index = getIndex(), pad = getPad())
index = getIndex(), pad = getPad(),
dtype = currentBatch.dtype, layout = currentBatch.layout)
} else {
currentBatch = null
}
Expand Down

0 comments on commit 9332064

Please sign in to comment.