dataset reset

python/jittor/__init__.py

@@ -7,7 +7,7 @@
 # This file is subject to the terms and conditions defined in
 # file 'LICENSE.txt', which is part of this source code package.
 # ***************************************************************
-__version__ = '1.2.2.9'
+__version__ = '1.2.2.10'
 from . import lock
 with lock.lock_scope():
     ori_int = int

python/jittor/dataset/dataset.py

@@ -114,6 +114,7 @@ class Dataset(object):
         for k,v in kw.items():
             assert hasattr(self, k), k
             setattr(self, k, v)
+        self.reset()
         return self
 
     def to_jittor(self, batch):