Explicit directory when saving

BrandonHanx · Dec 31, 2020 · 629e9ed · 629e9ed
1 parent ab1e43f
commit 629e9ed
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/encoding/utils/files.py b/encoding/utils/files.py
@@ -8,12 +8,8 @@
 
 __all__ = ['save_checkpoint', 'download', 'mkdir', 'check_sha1']
 
-def save_checkpoint(state, args, is_best, filename='checkpoint.pth.tar'):
+def save_checkpoint(state, directory, is_best, filename='checkpoint.pth.tar'):
     """Saves checkpoint to disk"""
-    if hasattr(args, 'backbone'):
-        directory = "runs/%s/%s/%s/%s/"%(args.dataset, args.model, args.backbone, args.checkname)
-    else:
-        directory = "runs/%s/%s/%s/"%(args.dataset, args.model, args.checkname)
     if not os.path.exists(directory):
         os.makedirs(directory)
     filename = directory + filename