henry000 commited on
Commit
7e84a57
·
1 Parent(s): a33e03b

💬 [Update] logger with quite mode

Browse files
Files changed (1) hide show
  1. yolo/utils/logging_utils.py +3 -1
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,