add level name to log

Xiao-Chenguang · Oct 22, 2024 · 4f8f6fc · 4f8f6fc
1 parent 43146c4
commit 4f8f6fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/fedmind/client.py b/fedmind/client.py
@@ -121,7 +121,7 @@ def train_process(
     """
     logging.basicConfig(
         level=log_level,
-        format="%(asctime)s [%(processName)s] %(message)s",
+        format="%(asctime)s %(levelname)s [%(processName)s] %(message)s",
     )
     logger = logging.getLogger(f"Worker-{worker_id}")
     logger.info(f"Worker-{worker_id} started.")

diff --git a/fedmind/server.py b/fedmind/server.py
@@ -56,7 +56,7 @@ def __init__(
 
         logging.basicConfig(
             level=args.LOG_LEVEL,  # type: ignore
-            format="%(asctime)s [%(processName)s] %(message)s",
+            format="%(asctime)s %(levelname)s [%(processName)s] %(message)s",
         )
         self.logger = logging.getLogger("Server")
         self.logger.info(f"Get following configs:\n{yaml.dump(args.to_dict())}")