mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Try copytree for tb_logger again.

Browse Source
This commit is contained in:
James Betker 2020-08-26 17:55:02 -06:00
parent cbd5e7a986
commit 80aa83bfd2
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/train.py
View File

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