multi datasets

2026-08-12 12:30:35 +08:00 · 2020-10-18 15:33:14 +08:00
parent 4453b25000
commit 6647ef5e65
3 changed files with 1266 additions and 1752 deletions
@@ -115,7 +115,7 @@ class Seq2SeqDataSets(torch.utils.data.Dataset):
        for d in self.datasets:
            l += len(d)
            if i < l:
-                return d[i]
+                return d[l-i]
        raise IndexError

    def get_rows(self, i):
@@ -123,7 +123,7 @@ class Seq2SeqDataSets(torch.utils.data.Dataset):
        for d in self.datasets:
            l += len(d)
            if i < l:
-                return d.get_rows(i)
+                return d.get_rows(l-i)
        raise IndexError

    def __len__(self):