dataset to_jittor recursive

2020-12-25 16:08:13 +08:00 · 2020-12-25 16:08:13 +08:00 · cd085765a0
parent 2c939cb301
commit cd085765a0
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/dataset/dataset.py
+++ b/python/jittor/dataset/dataset.py
@ -136,7 +136,7 @@ class Dataset(object):
                isinstance(a, float):
                new_batch.append(to_jt(a))
            else:
-                new_batch.append(a)
+                new_batch.append(self.to_jittor(a))
        return new_batch

    def collate_batch(self, batch):