22 lines
745 B
Python
22 lines
745 B
Python
import os
|
|
import logging
|
|
|
|
from torch.utils.tensorboard import SummaryWriter
|
|
|
|
def get_logger(filename=None):
|
|
logger = logging.getLogger('logger')
|
|
logger.setLevel(logging.DEBUG)
|
|
logging.basicConfig(format='%(asctime)s - %(levelname)s - %(message)s',
|
|
datefmt='%m/%d/%Y %H:%M:%S',
|
|
level=logging.INFO)
|
|
if filename is not None:
|
|
handler = logging.FileHandler(filename)
|
|
handler.setLevel(logging.DEBUG)
|
|
handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s: %(message)s'))
|
|
logging.getLogger().addHandler(handler)
|
|
return logger
|
|
|
|
def get_summary_writer(dirname: str):
|
|
|
|
os.makedirs(dirname, exist_ok=True)
|
|
return SummaryWriter(log_dir=dirname) |