# mylog.py
import logging
import os
import sys
def init_logging(log_file, stdout=False):
formatter = logging.Formatter('%(asctime)s [%(levelname)s] %(module)s: %(message)s',
datefmt='%m/%d/%Y %H:%M:%S')
print('Making log output file: %s' % log_file)
print(log_file[: log_file.rfind(os.sep)])
if not os.path.exists(log_file[: log_file.rfind(os.sep)]):
os.makedirs(log_file[: log_file.rfind(os.sep)])
fh = logging.FileHandler(log_file)
fh.setFormatter(formatter)
fh.setLevel(logging.INFO)
logger = logging.getLogger()
logger.addHandler(fh)
logger.setLevel(logging.INFO)
if stdout:
ch = logging.StreamHandler(sys.stdout)
ch.setFormatter(formatter)
ch.setLevel(logging.INFO)
logger.addHandler(ch)
return logger
#main.py
import logging
import mylog
logging = mylog.init_logging('output.log', stdout=True)
logging.info('Parameters:')
输出:11/17/2021 14:28:28 [INFO] train_demo: Parameters: