Add debug logging to worker_training message reception

parent 9344ec30
...@@ -60,6 +60,8 @@ def worker_process(backend_type: str): ...@@ -60,6 +60,8 @@ def worker_process(backend_type: str):
while True: while True:
try: try:
message = comm.receive_message() message = comm.receive_message()
if message:
print(f"DEBUG: Worker {os.getpid()} received message: {message}")
if message and message.msg_type == 'train_request': if message and message.msg_type == 'train_request':
print(f"DEBUG: Worker received train_request: {message.msg_id}") print(f"DEBUG: Worker received train_request: {message.msg_id}")
data = message.data data = message.data
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment