deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tblogger..

Browse Source
This commit is contained in:
James Betker 2020-04-30 12:35:51 -06:00
parent b6e036147a
commit 03258445bc
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/train.py
View File

@ -73,9 +73,10 @@ def main():
logger.info(option.dict2str(opt))
# tensorboard logger
if opt['use_tb_logger'] and 'debug' not in opt['name']:
tb_logger_path = '../tb_logger/' + opt['name']
# If not resuming, delete the existing logs. Tensorboard doesn't do too great with these.
if 'resume_state' not in opt['path'].keys():
tb_logger_path = '../tb_logger/' + opt['name']
if opt['path'].get('resume_state', None) is None:
if os.path.exists(tb_logger_path) and os.path.isdir(tb_logger_path):
shutil.rmtree(tb_logger_path)