Print current time in launch_bps

ruipeterpan · Jan 18, 2021 · 16061e9 · 16061e9
1 parent f6c063e
commit 16061e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/launcher/launch.py b/launcher/launch.py
@@ -7,6 +7,7 @@
 import threading
 import sys
 import time
+from datetime import datetime
 from functools import reduce
 
 
@@ -178,7 +179,7 @@ def worker(local_rank, local_size, command, allocation=None):
 
 
 def launch_bps():
-    print("BytePS launching " + os.environ["DMLC_ROLE"])
+    print(f"[{datetime.now().strftime("%H:%M:%S")}] BytePS launching {os.environ["DMLC_ROLE"]}")
     sys.stdout.flush()
     check_env()
     if os.environ["DMLC_ROLE"] == "worker":