Add debug logging to worker_training message reception

parent 9344ec30
......@@ -60,6 +60,8 @@ def worker_process(backend_type: str):
while True:
try:
message = comm.receive_message()
if message:
print(f"DEBUG: Worker {os.getpid()} received message: {message}")
if message and message.msg_type == 'train_request':
print(f"DEBUG: Worker received train_request: {message.msg_id}")
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