wassname/DeepRL
1
0
Fork 0
mirror of https://github.com/wassname/DeepRL.git synced 2026-09-12 12:05:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove PIL

Browse Source
This commit is contained in:
Shangtong Zhang
2018-02-09 16:34:30 -07:00
parent d120cf44d9
commit b24593352d
5 changed files with 47 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/tf_logger.py
+17 -3
View File
@@ -5,21 +5,35 @@
#######################################################################
from tensorboardX import SummaryWriter
import os
import numpy as np
class Logger(object):
def __init__(self, log_dir, vanilla_logger, skip=False):
self.writer = SummaryWriter(log_dir)
for f in os.listdir(log_dir):
os.remove('%s/%s' % (log_dir, f))
if not skip:
self.writer = SummaryWriter(log_dir)
self.info = vanilla_logger.info
self.debug = vanilla_logger.debug
self.warning = vanilla_logger.warning
self.skip = skip
self.step = 0
def scalar_summary(self, tag, value, step):
def scalar_summary(self, tag, value, step=None):
if self.skip:
return
if step is None:
step = self.step
self.step += 1
if np.isscalar(value):
value = np.asarray([value])
self.writer.add_scalar(tag, value, step)
def histo_summary(self, tag, values, step):
def histo_summary(self, tag, values, step=None):
if self.skip:
return
if step is None:
step = self.step
self.step += 1
self.writer.add_histogram(tag, values, step, bins=1000)