import logging def set_logger(log_path): """Set the logger to log info in terminal and file `log_path`. ``` logging.info("Starting training...") ``` Args: log_path: (string) where to log """ logger = logging.getLogger(__name__) logger.setLevel(logging.INFO) logger.propagate = False # Logging to a file file_handler = logging.FileHandler(log_path) file_handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s: %(message)s')) logger.addHandler(file_handler) # Logging to console stream_handler = logging.StreamHandler() stream_handler.setFormatter(logging.Formatter('%(message)s')) logger.addHandler(stream_handler) return logger