💬 [Update] logger with quite mode
Browse files
yolo/utils/logging_utils.py
CHANGED
@@ -27,8 +27,10 @@ from torch.optim import Optimizer
|
|
27 |
from yolo.config.config import Config, YOLOLayer
|
28 |
|
29 |
|
30 |
-
def custom_logger():
|
31 |
logger.remove()
|
|
|
|
|
32 |
logger.add(
|
33 |
sys.stderr,
|
34 |
colorize=True,
|
|
|
27 |
from yolo.config.config import Config, YOLOLayer
|
28 |
|
29 |
|
30 |
+
def custom_logger(quite: bool = False):
|
31 |
logger.remove()
|
32 |
+
if quite:
|
33 |
+
return
|
34 |
logger.add(
|
35 |
sys.stderr,
|
36 |
colorize=True,
|