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
|