generate the right number of dmeos

mattdeitke · Feb 1, 2019 · 7778825 · 7778825
1 parent 69ea2b3
commit 7778825
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/scripts/make_agent_demos.py b/scripts/make_agent_demos.py
@@ -81,15 +81,16 @@ def generate_demos(n_episodes, valid, seed, shift=0):
     checkpoint_time = time.time()
 
     just_crashed = False
-    for i in range(n_episodes):
-        # Run the expert for one episode
+    while True:
+        if len(demos) == n_episodes:
+            break
 
         done = False
         if just_crashed:
             logger.info("reset the environment to find a mission that the bot can solve")
             env.reset()
         else:
-            env.seed(seed + i)
+            env.seed(seed + len(demos))
         obs = env.reset()
         agent.on_reset()