From a3a09a48376b6d332596256c2ec16bf942fe1c91 Mon Sep 17 00:00:00 2001
From: guoye yang <498731903@qq.com>
Date: Thu, 7 Jan 2021 10:23:09 +0800
Subject: [PATCH] fix

---
 python/jittor/dataset/dataset.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/jittor/dataset/dataset.py b/python/jittor/dataset/dataset.py
index 4d0160fd..774779af 100644
--- a/python/jittor/dataset/dataset.py
+++ b/python/jittor/dataset/dataset.py
@@ -128,7 +128,8 @@ class Dataset(object):
             if self.stop_grad else jt.array(x)
         if isinstance(batch, np.ndarray):
             return to_jt(batch)
-        assert isinstance(batch, Sequence)
+        if not isinstance(batch, (list, tuple)):
+            return batch
         new_batch = []
         for a in batch:
             if isinstance(a, np.ndarray) or \