Merge pull request #169 from Jittor/dataset

fix dataset to_jittor bug
2020-12-16 15:57:46 +08:00 · 2020-12-16 15:57:46 +08:00 · cf8ba20fe3
parent f9be78c1d4 23de860d68
commit cf8ba20fe3
1 changed files with 1 additions and 0 deletions
--- a/python/jittor/dataset/dataset.py
+++ b/python/jittor/dataset/dataset.py
@ -117,6 +117,7 @@ class Dataset(object):
        '''
        Change batch data to jittor array, such as np.ndarray, int, and float.
        '''
+        if isinstance(batch, jt.Var): return batch
        to_jt = lambda x: jt.array(x).stop_grad() \
            if self.stop_grad else jt.array(x)
        if isinstance(batch, np.ndarray):