changed timer sleep

luigiba · Sep 26, 2019 · 7533439 · 7533439
1 parent 47f9c20
commit 7533439
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/distribute_training.py b/distribute_training.py
@@ -365,9 +365,13 @@ def main_fun(argv, ctx):
 
             #gives time to chief to stop
             if task_index != 0:
-                time.sleep(15)
+                time.sleep(30)
 
         else:
+            #gives time to other workers to connect
+            if task_index == 0:
+                time.sleep(30)
+
             d = {
                 'r_tot' : 0.0, 'r_filter_tot' : 0.0, 'r_tot_constrain' : 0.0, 'r_filter_tot_constrain' : 0.0,
                 'r1_tot' : 0.0, 'r1_filter_tot' : 0.0, 'r1_tot_constrain' : 0.0, 'r1_filter_tot_constrain' : 0.0,
@@ -603,7 +607,9 @@ def main_fun(argv, ctx):
                     f.write(str(key) + ":" + str(d[key])+'\n')
 
 
-
+            #gives time to chief to stop
+            if task_index != 0:
+                time.sleep(30)