future-xy
commited on
Commit
·
a4a186c
1
Parent(s):
b4759e0
set log level to warning
Browse files- backend-cli.py +13 -2
- src/backend/run_eval_suite.py +1 -0
backend-cli.py
CHANGED
|
@@ -25,10 +25,21 @@ from typing import Optional
|
|
| 25 |
|
| 26 |
import time
|
| 27 |
|
| 28 |
-
import logging
|
| 29 |
import pprint
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
|
|
|
|
| 32 |
eval_logger.setLevel(logging.WARNING)
|
| 33 |
|
| 34 |
|
|
|
|
| 25 |
|
| 26 |
import time
|
| 27 |
|
|
|
|
| 28 |
import pprint
|
| 29 |
+
import logging
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
# Configure the root logger
|
| 33 |
+
logging.basicConfig(
|
| 34 |
+
format="%(asctime)s,%(msecs)03d %(levelname)-8s [%(filename)s:%(lineno)d] %(message)s",
|
| 35 |
+
datefmt="%Y-%m-%d:%H:%M:%S",
|
| 36 |
+
level=logging.WARNING,
|
| 37 |
+
)
|
| 38 |
+
|
| 39 |
+
# Get the 'lm-eval' logger from the third-party library
|
| 40 |
+
eval_logger = logging.getLogger("lm-eval")
|
| 41 |
|
| 42 |
+
# Explicitly set the level for 'lm-eval' logger to WARNING
|
| 43 |
eval_logger.setLevel(logging.WARNING)
|
| 44 |
|
| 45 |
|
src/backend/run_eval_suite.py
CHANGED
|
@@ -59,6 +59,7 @@ def run_evaluation(
|
|
| 59 |
limit=limit,
|
| 60 |
write_out=True,
|
| 61 |
task_manager=task_manager,
|
|
|
|
| 62 |
)
|
| 63 |
|
| 64 |
results["config"]["model_dtype"] = eval_request.precision
|
|
|
|
| 59 |
limit=limit,
|
| 60 |
write_out=True,
|
| 61 |
task_manager=task_manager,
|
| 62 |
+
verbosity="WARNING",
|
| 63 |
)
|
| 64 |
|
| 65 |
results["config"]["model_dtype"] = eval_request.precision
|