Update zk.py

zk.py

@@ -8,7 +8,7 @@ import sys
 def parse_args():
     parser = argparse.ArgumentParser(description='Matrix multiplication')
     parser.add_argument('--gpus', help='gpu amount',default = 1,  type=int)
-    parser.add_argument('--size', help='matrix size', default =40000, type=int)
+    parser.add_argument('--size', help='matrix size', default =2000, type=int)
     parser.add_argument('--interval', help='sleep interval', default = 0.01, type=float)
     args = parser.parse_args()
     return args