update comments

Author: irenedea

rollout.py

@@ -70,8 +70,9 @@
         # TODO: start generating rollouts for the experience buffer
 
         # Send the experience buffer to the train agent.
+        # We do not block here. We can continue generating rollouts while the experience buffer is being sent.
         experience_buffer = torch.tensor([20+i])
-        experience_buffer_work = torch.distributed.broadcast(group=experience_buffer_group, src=1,tensor=experience_buffer, async_op=True) # don't block, send it off and continue generating rollouts
+        experience_buffer_work = torch.distributed.broadcast(group=experience_buffer_group, src=1,tensor=experience_buffer, async_op=True)
         log.info(f"Sent experience buffer {experience_buffer}")
 
         log.info(f"Completed iteration {i + 1}/{MAX_ITERATIONS}")