You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by cj...@apache.org on 2017/11/05 20:58:11 UTC
[incubator-mxnet] 02/05: fix gluon.data.RecordFileDataset (#8353)
This is an automated email from the ASF dual-hosted git repository.
cjolivier01 pushed a commit to branch v0.12.0
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
commit 8d5001be9c784658e6f5ef9f8d921a7c6e7d0d25
Author: Joshua Z. Zhang <ch...@gmail.com>
AuthorDate: Tue Oct 24 21:19:00 2017 -0700
fix gluon.data.RecordFileDataset (#8353)
---
python/mxnet/gluon/data/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/mxnet/gluon/data/dataset.py b/python/mxnet/gluon/data/dataset.py
index 28c6595..cbc73bc 100644
--- a/python/mxnet/gluon/data/dataset.py
+++ b/python/mxnet/gluon/data/dataset.py
@@ -81,7 +81,7 @@ class RecordFileDataset(Dataset):
self._record = recordio.MXIndexedRecordIO(idx_file, filename, 'r')
def __getitem__(self, idx):
- return self._record.read_idx(idx)
+ return self._record.read_idx(self._record.keys[idx])
def __len__(self):
return len(self._record.keys)
--
To stop receiving notification emails like this one, please contact
"commits@mxnet.apache.org" <co...@mxnet.apache.org>.