瀏覽代碼

clearer startup logs

justheuristic 2 年之前
父節點
當前提交
2e93bd7139
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/server/server.py

+ 1 - 1
src/server/server.py

@@ -73,7 +73,7 @@ class Server(threading.Thread):
         logger.info(f"Serving {len(self.module_backends)} blocks:")
         for block_name, backend in self.module_backends.items():
             num_parameters = sum(p.numel() for p in backend.module.parameters() if p.requires_grad)
-            parameter_msg = f'{num_parameters} parameters' if num_parameters else "frozen"
+            parameter_msg = f'{num_parameters} trainable parameters' if num_parameters else "frozen"
             logger.info(f"{block_name}: {backend.module.__class__.__name__}, {parameter_msg}")
 
         if not self.dht.is_alive():