File size: 1,564 Bytes
f60c555 |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import logging
import os
def init_logger(config):
"""
A logger that can show a message on standard output and write it into the
file named `filename` simultaneously.
All the message that you want to log MUST be str.
Args:
config (Config): An instance object of Config, used to record parameter information.
"""
LOGROOT = './log/'
dir_name = os.path.dirname(LOGROOT)
if not os.path.exists(dir_name):
os.makedirs(dir_name)
logger = logging.getLogger("normal")
name_ = "{}-{}-lr_{}-rww_{}-nl_{}-sdp_{}-mdp_{}-clt_{}-diffw_{}-semw_{}.log"
logfilename = name_.format(config.model_name, config.dataset, config.learning_rate,
config.reg_weight, config.n_layers,
config.s_drop, config.m_drop, config.cl_tmp,
config.diff_loss_weight, config.cl_loss_weight)
logfilepath = os.path.join(LOGROOT, logfilename)
filefmt = "%(asctime)-15s %(message)s"
filedatefmt = "%a %d %b %Y %H:%M:%S"
fileformatter = logging.Formatter(filefmt, filedatefmt)
sfmt = u"%(asctime)-15s %(message)s"
sdatefmt = "%d %b %H:%M"
sformatter = logging.Formatter(sfmt, sdatefmt)
fh = logging.FileHandler(logfilepath, 'w', 'utf-8')
fh.setFormatter(fileformatter)
sh = logging.StreamHandler()
sh.setFormatter(sformatter)
logger.setLevel(logging.INFO)
logger.handlers = []
logger.addHandler(fh)
logger.addHandler(sh)
return logger
|