Update app.py
Browse files
app.py
CHANGED
@@ -148,6 +148,7 @@ def test(base_model_id, dataset):
|
|
148 |
handlers=[logging.StreamHandler(sys.stdout)],
|
149 |
)
|
150 |
log_level = train_conf.get_process_log_level()
|
|
|
151 |
logger.setLevel(log_level)
|
152 |
datasets.utils.logging.set_verbosity(log_level)
|
153 |
transformers.utils.logging.set_verbosity(log_level)
|
|
|
148 |
handlers=[logging.StreamHandler(sys.stdout)],
|
149 |
)
|
150 |
log_level = train_conf.get_process_log_level()
|
151 |
+
logger = logging.getLogger("FT")
|
152 |
logger.setLevel(log_level)
|
153 |
datasets.utils.logging.set_verbosity(log_level)
|
154 |
transformers.utils.logging.set_verbosity(log_level)
|