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

Merge branch 'master' of https://github.com/ShangtongZhang/DeepRL

Browse Source
This commit is contained in:
Shangtong Zhang
2018-03-13 09:20:26 -06:00
parent d00c3b6829 1191292c51
commit 174cc50440
3 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/tf_logger.py
+5 -2
View File
@@ -10,8 +10,11 @@ import numpy as np
class Logger(object):
def __init__(self, log_dir, vanilla_logger, skip=False):
for f in os.listdir(log_dir):
os.remove('%s/%s' % (log_dir, f))
try:
for f in os.listdir(log_dir):
os.remove('%s/%s' % (log_dir, f))
except FileNotFoundError:
os.mkdir(log_dir)
if not skip:
self.writer = SummaryWriter(log_dir)
self.info = vanilla_logger.info