polish error msg

python/jittor/__init__.py

@@ -8,7 +8,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.27'
+__version__ = '1.2.2.28'
 from . import lock
 with lock.lock_scope():
     ori_int = int

python/jittor/dataset/dataset.py

@@ -392,7 +392,7 @@ Example::
             self.real_len = len(index_list)
             self.real_batch_size = real_batch_size
             assert self.total_len // self.batch_size == \
-                self.real_len // self.real_batch_size
+                self.real_len // self.real_batch_size, f"Number of batches({self.total_len // self.batch_size}!={self.real_len // self.real_batch_size}) not match, total_len: {self.total_len}, batch_size: {self.batch_size}, real_len: {self.real_len}, real_batch_size: {self.real_batch_size}"
         else:
             self.real_len = self.total_len
             self.real_batch_size = self.batch_size

python/jittor/models/alexnet.py

@@ -23,7 +23,7 @@ class AlexNet(nn.Module):
     Example::
     
         model = jittor.models.AlexNet(500)
-        x = jittor.random([10,224,224,3])
+        x = jittor.random([10,3,224,224])
         y = model(x) # [10, 500]
 
     """