Increase tests timeout and reduce iterations in parameter_server example

Author: adityapb

auto_test.py

@@ -30,7 +30,7 @@ def searchForPython(python_implementations):
 
 
 # ----------------------------------------------------------------------------------
-TIMEOUT = 60  # timeout for each test (in seconds)
+TIMEOUT = 120  # timeout for each test (in seconds)
 CHARM_QUIET_AFTER_NUM_TESTS = 5
 
 commonArgs = ['++local']

examples/ray/parameter_server.py

@@ -125,7 +125,7 @@ def compute_gradients(self, weights):
 def sync_train(args):
     ray.init()
 
-    iterations = 101
+    iterations = 11
     num_workers = 4
 
     model = ConvNet()
@@ -159,7 +159,7 @@ def async_train(args):
 
     model = ConvNet()
     test_loader = get_data_loader()[1]
-    iterations = 101
+    iterations = 11
     num_workers = 4
 
     print("Running asynchronous parameter server training.")