Update DataLoader.cs

yueyinqiu · yueyinqiu · commit 2a9344a7673e · 2024-05-08T23:05:57.000+08:00
diff --git a/src/TorchSharp/DataLoader.cs b/src/TorchSharp/DataLoader.cs
@@ -319,6 +319,8 @@ public bool MoveNext()
                             .Where(x => x.HasValue)
                             .Cast<long>()
                             .ToArray();
+                        if (indices.Length is 0)
+                            return false;
                         if (loader.drop_last && indices.Length < loader.batchSize) {
                             return false;
                         }

Original file line number	Diff line number	Diff line change
`@@ -319,6 +319,8 @@ public bool MoveNext()`
`319`	`319`	`.Where(x => x.HasValue)`
`320`	`320`	`.Cast<long>()`
`321`	`321`	`.ToArray();`
	`322`	`+ if (indices.Length is 0)`
	`323`	`+ return false;`
`322`	`324`	`if (loader.drop_last && indices.Length < loader.batchSize) {`
`323`	`325`	`return false;`
`324`	`326`	`}`