File size: 519 Bytes
72f4d4d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import logging
import os
def setup_logger(name="train_log", log_file="logs/training.log"):
os.makedirs(os.path.dirname(log_file), exist_ok=True)
logger = logging.getLogger(name)
logger.setLevel(logging.INFO)
fh = logging.FileHandler(log_file)
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
fh.setFormatter(formatter)
logger.addHandler(fh)
return logger
if __name__ == "__main__":
log = setup_logger()
log.info("Logger initialized.")
|