more on NVlabs#4

hma02 · May 30, 2017 · 8f138bd · 8f138bd
1 parent f929bd7
commit 8f138bd
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 6 deletions.
diff --git a/ga3c/ProcessStats.py b/ga3c/ProcessStats.py
@@ -70,7 +70,7 @@ def run(self):
             first_time = datetime.now()
             while self.exit_flag.value == 0:
                 try:
-                    episode_time, reward, length = self.episode_log_q.get(timeout=2)
+                    episode_time, reward, length = self.episode_log_q.get(timeout=0.1)
                 except:
                     continue
 

diff --git a/ga3c/Server.py b/ga3c/Server.py
@@ -122,7 +122,11 @@ def main(self):
                 self.stats.should_save_model.value = 0
 
             time.sleep(0.01)
-
+
+        print('Finished. Exiting subprocesses ...')
+
+        self.stats.exit_flag.value = True
+        self.stats.join()
         self.dynamic_adjustment.exit_flag = True
         self.dynamic_adjustment.join()
         while self.agents:
@@ -132,6 +136,5 @@ def main(self):
         while self.trainers:
             self.remove_trainer()
 
-        self.stats.exit_flag.value = True
-        self.stats.join()
+
 
diff --git a/ga3c/ThreadPredictor.py b/ga3c/ThreadPredictor.py
@@ -48,7 +48,7 @@ def run(self):
 
         while not self.exit_flag:
             try:
-                ids[0], states[0] = self.server.prediction_q.get(timeout=2)
+                ids[0], states[0] = self.server.prediction_q.get(timeout=0.1)
             except:
                 continue
 

diff --git a/ga3c/ThreadTrainer.py b/ga3c/ThreadTrainer.py
@@ -45,7 +45,7 @@ def run(self):
             batch_size = 0
             while batch_size <= Config.TRAINING_MIN_BATCH_SIZE:
                 try:
-                    x_, r_, a_ = self.server.training_q.get(timeout=2)
+                    x_, r_, a_ = self.server.training_q.get(timeout=0.1)
                 except:
                     if self.exit_flag: break
                     continue