more printing

FilipinoGambino · Feb 8, 2024 · 0a80aee · 0a80aee
1 parent d3d72f4
commit 0a80aee
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/connectx/torchbeast/monobeast.py b/connectx/torchbeast/monobeast.py
@@ -197,11 +197,13 @@ def act(
     try:
         logging.info(f"Actor {actor_index} started.")
         timings = prof.Timings()
-
+        logging.info(f"creating env")
         env = create_env(flags, device=flags.actor_device, teacher_flags=teacher_flags)
-
+        logging.info(f"getting env output")
         env_output = env.reset(force=True)
+        logging.info(f"getting agent output")
         agent_output = actor_model(env_output)
+        logging.info(f"beginning loop")
         while True:
             index = free_queue.get()
             if index is None: