log_dir needs root prefix

2019-01-09 18:06:04 +00:00 · 2019-01-09 18:06:04 +00:00 · eed545bbf7
parent 81b7ea7e72
commit eed545bbf7
1 changed files with 1 additions and 1 deletions
--- a/arguments.py
+++ b/arguments.py
@ -132,7 +132,7 @@ def parse():
    args.dist_sync_file = os.path.join(args.log_dir, 'distributed_sync_file')
    
    save_args(args)
-    for x in ['data', 'save', 'embeddings']:
+    for x in ['data', 'save', 'embeddings', 'log_dir']:
        setattr(args, x, os.path.join(args.root, getattr(args, x)))

    return args