lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

This commit is contained in:
guoye yang 2021-01-07 10:23:09 +08:00
parent d80e4056f6
commit a3a09a4837
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python/jittor/dataset/dataset.py
View File

@ -128,7 +128,8 @@ class Dataset(object):
if self.stop_grad else jt.array(x) if self.stop_grad else jt.array(x)
if isinstance(batch, np.ndarray): if isinstance(batch, np.ndarray):
return to_jt(batch) return to_jt(batch)
assert isinstance(batch, Sequence) if not isinstance(batch, (list, tuple)):
return batch
new_batch = [] new_batch = []
for a in batch: for a in batch:
if isinstance(a, np.ndarray) or \ if isinstance(a, np.ndarray) or \