mirror of
https://github.com/wassname/Run-Skeleton-Run.git
synced 2026-06-27 18:21:58 +08:00
wassname
23 lines
674 B
Python
23 lines
674 B
Python
from tensorboardX import SummaryWriter
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class Logger(object):
|
|
def __init__(self, log_dir, vanilla_logger=logger, skip=False):
|
|
self.writer = SummaryWriter(log_dir)
|
|
self.info = vanilla_logger.info
|
|
self.debug = vanilla_logger.debug
|
|
self.warning = vanilla_logger.warning
|
|
self.skip = skip
|
|
|
|
def scalar_summary(self, tag, value, step):
|
|
if self.skip:
|
|
return
|
|
self.writer.add_scalar(tag, value, step)
|
|
|
|
def histo_summary(self, tag, values, step):
|
|
if self.skip:
|
|
return
|
|
self.writer.add_histogram(tag, values, step, bins=1000)
|