File size: 417 Bytes
0174b5b 91a3f78 0174b5b ca2b494 0174b5b 91a3f78 802cb12 |
1 2 3 4 5 6 7 8 9 10 11 12 |
import logging
from lightning.pytorch.utilities.rank_zero import rank_zero_only
from rich.console import Console
from rich.logging import RichHandler
logger = logging.getLogger("yolo")
logger.setLevel(logging.DEBUG)
logger.propagate = False
if rank_zero_only.rank == 0 and not logger.hasHandlers():
logger.addHandler(RichHandler(console=Console(), show_level=True, show_path=True, show_time=True, markup=True))
|