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

Stupid pathing..

Browse Source
This commit is contained in:
James Betker 2020-08-26 17:58:24 -06:00
parent 25832930db
commit 623f3b99b2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/train.py
View File

@ -280,9 +280,9 @@ def main():
if 'alt_path' in opt['path'].keys(): if 'alt_path' in opt['path'].keys():
import shutil import shutil
print("Synchronizing tb_logger to alt_path..") print("Synchronizing tb_logger to alt_path..")
alt_tblogger = os.path.join(tb_logger_path, "tb_logger") alt_tblogger = os.path.join(opt['path']['alt_path'], "tb_logger")
shutil.rmtree(alt_tblogger, ignore_errors=True) shutil.rmtree(alt_tblogger, ignore_errors=True)
shutil.copytree(alt_tblogger, opt['path']['alt_path']) shutil.copytree(tb_logger_path, alt_tblogger)
if rank <= 0: if rank <= 0:
logger.info('Saving the final model.') logger.info('Saving the final model.')