diff --git a/python/paddle/fluid/distributed/ps_instance.py b/python/paddle/fluid/distributed/ps_instance.py
index 61b2bcad01d5b..42033a0ada4ac 100644
--- a/python/paddle/fluid/distributed/ps_instance.py
+++ b/python/paddle/fluid/distributed/ps_instance.py
@@ -24,7 +24,7 @@ class PaddlePSInstance(object):
             instance = PaddlePSInstance(1, 2)
     """
 
-    def __init__(self, server_worker_mode, proc_per_node):
+    def __init__(self, server_worker_mode=1, proc_per_node=2):
         self.dh = MPIHelper()
         self._rankid = self.dh.get_rank()
         self._server_worker_mode = server_worker_mode