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.")