28 lines
725 B
Python
28 lines
725 B
Python
|
|
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
|