fixed dutil/Dataset.Next()

CHANGELOG.md

@@ -6,6 +6,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
 ## [Unreleased]
+- Fixed incorrect indexing at `dutil/Dataset.Next()`
 - Added `nn.MSELoss()`
 
 ## [Nofix]

dutil/dataloader.go

@@ -98,7 +98,7 @@ func (dl *DataLoader) Next() (interface{}, error) {
 		nextIndex = len(dl.indexes)
 	}
 	for i := dl.currIdx; i < nextIndex; i++ {
-		item, err := dl.dataset.Item(i)
+		item, err := dl.dataset.Item(dl.indexes[i])
 		if err != nil {
 			return nil, err
 		}