Update LunarLander-v2_PPO.py

pythonlessons · Nov 12, 2020 · db9783a · db9783a
1 parent 2222a20
commit db9783a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/LunarLander-v2_PPO/LunarLander-v2_PPO.py b/LunarLander-v2_PPO/LunarLander-v2_PPO.py
@@ -321,6 +321,7 @@ def run(self): # train only when episode is finished
 
             if self.episode >= self.EPISODES:
                 break
+        self.env.close()
 
     def run_batch(self): # train every self.Training_batch episodes
         state = self.env.reset()
@@ -358,7 +359,9 @@ def run_batch(self): # train every self.Training_batch episodes
             self.replay(states, actions, rewards, predictions, dones, next_states)
             if self.episode >= self.EPISODES:
                 break
+        self.env.close()  
 
+
     def run_multiprocesses(self, num_worker = 4):
         works, parent_conns, child_conns = [], [], []
         for idx in range(num_worker):