File size: 711 Bytes
408f9e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import os
import logging
_format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
def config_logger(log_level=logging.INFO):
logging.basicConfig(format=_format, level=log_level)
def log_to_file(logger_name=None, log_level=logging.INFO, log_filename='tensorflow.log'):
if not os.path.exists(os.path.dirname(log_filename)):
os.makedirs(os.path.dirname(log_filename))
if logger_name is not None:
log = logging.getLogger(logger_name)
else:
log = logging.getLogger()
fh = logging.FileHandler(log_filename)
fh.setLevel(log_level)
fh.setFormatter(logging.Formatter(_format))
log.addHandler(fh)
def log_versions():
import torch
import subprocess
|