diff --git a/function.py b/function.py
index 20194d8..a81f5b1 100644
--- a/function.py
+++ b/function.py
@@ -28,7 +28,7 @@ def sigmoid(z):
     return 1.0 / (1 + np.exp(-z))
 
 # 这里写自己的代码实现
-def train_logreg(X, y, learning_rate=0.01, iterations=1000, batch_size=128):
+def train_logreg(X, y, learning_rate=0.01, iterations=1000, batch_size=256):
     """
     对率回归训练函数。