resnet tune batch size

python/jittor/__init__.py

@@ -9,7 +9,7 @@
 # file 'LICENSE.txt', which is part of this source code package.
 # ***************************************************************
 
-__version__ = '1.2.3.33'
+__version__ = '1.2.3.34'
 from jittor_utils import lock
 with lock.lock_scope():
     ori_int = int

python/jittor/test/test_resnet.py

@@ -38,7 +38,7 @@ class TestResnet(unittest.TestCase):
     @classmethod
     def setUpClass(self):
         # hyper-parameters
-        self.batch_size = 100
+        self.batch_size = int(os.environ.get("TEST_BATCH_SIZE", "100"))
         self.weight_decay = 0.0001
         self.momentum = 0.9
         self.learning_rate = 0.1