From 0a80aee580d746026c3edece6471684312a4a679 Mon Sep 17 00:00:00 2001
From: FilipinoGambino <ngorichs1@gmail.com>
Date: Wed, 7 Feb 2024 20:22:24 -0600
Subject: [PATCH] more printing

---
 connectx/torchbeast/monobeast.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/connectx/torchbeast/monobeast.py b/connectx/torchbeast/monobeast.py
index 25162bd..b47c008 100644
--- 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: