be verbose when saving demos

bruehldev · Feb 4, 2019 · 07f2133 · 07f2133
1 parent b33ffc6
commit 07f2133
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/make_agent_demos.py b/scripts/make_agent_demos.py
@@ -141,15 +141,15 @@ def generate_demos(n_episodes, valid, seed, shift=0):
         if args.save_interval > 0 and len(demos) < n_episodes and len(demos) % args.save_interval == 0:
             logger.info("Saving demos...")
             utils.save_demos(demos, demos_path)
-            logger.info("Demos saved")
+            logger.info("{} demos saved".format(len(demos)))
             # print statistics for the last 100 demonstrations
             print_demo_lengths(demos[-100:])
 
 
     # Save demonstrations
     logger.info("Saving demos...")
     utils.save_demos(demos, demos_path)
-    logger.info("Demos saved")
+    logger.info("{} demos saved".format(len(demos)))
     print_demo_lengths(demos[-100:])