CAMP-VQA / src /utils /logger_setup.py
Xinyi Wang
initial commit
b9b1b10
raw
history blame contribute delete
733 Bytes
import logging
def setup_logging(log_file="log_debug.log"):
"""Setup logging to output to both console and file."""
logger = logging.getLogger()
logger.setLevel(logging.INFO)
# Console handler
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.INFO)
# File handler
file_handler = logging.FileHandler(log_file, mode='w', encoding='utf-8')
file_handler.setLevel(logging.INFO)
# Formatter for handlers
formatter = logging.Formatter('[%(asctime)s] %(levelname)s: %(message)s')
console_handler.setFormatter(formatter)
file_handler.setFormatter(formatter)
# Add handlers to logger
logger.addHandler(console_handler)
logger.addHandler(file_handler)