diff --git a/python/jittor/test/test_arg_pool_op.py b/python/jittor/test/test_arg_pool_op.py
index f8aac2fa..abf6fe81 100644
--- a/python/jittor/test/test_arg_pool_op.py
+++ b/python/jittor/test/test_arg_pool_op.py
@@ -81,9 +81,9 @@ class TestArgPoolOp(unittest.TestCase):
     def test_cuda(self):
         jt_model = jt.nn.Sequential(Pool(2, 2, 0), Pool(2, 2, 0), Pool(2, 2, 0, ceil_mode=True), Pool(2, 2, 0), Pool(2, 2, 0), Pool(3, 1, 1))
         torch_model = Sequential(MaxPool2d(2, 2, 0), MaxPool2d(2, 2, 0), MaxPool2d(2, 2, 0, ceil_mode=True), MaxPool2d(2, 2, 0), MaxPool2d(2, 2, 0), MaxPool2d(3, 1, 1))
-        shape = [64, 64, 300, 300]
+        shape = [2, 3, 300, 300]
         check(jt_model, torch_model, shape, False)
-        shape = [32, 128, 157, 300]
+        shape = [2, 3, 157, 300]
         check(jt_model, torch_model, shape, False)
         for i in range(10):
             check(jt_model, torch_model, [1,1,300,300], True)
diff --git a/python/jittor/utils/publish.py b/python/jittor/utils/publish.py
index c0c0841e..ad956677 100644
--- a/python/jittor/utils/publish.py
+++ b/python/jittor/utils/publish.py
@@ -10,6 +10,8 @@
 # Publish steps:
 # 1. build,push,upload docker image[jittor/jittor]
 # 2. build,push,upload docker image[jittor/jittor-cuda]
+# upload to pip:
+# rm -rf dist && python3.7 ./setup.py sdist && python3.7 -m twine upload dist/*
 import os
 
 def run_cmd(cmd):