RESUME with output the same as config dir

stefanklut · Jan 31, 2024 · 32c6834 · 32c6834
1 parent 36a9c13
commit 32c6834
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/core/setup.py b/core/setup.py
@@ -98,11 +98,10 @@ def setup_cfg(args, cfg: Optional[CfgNode] = None) -> CfgNode:
     _uuid = uuid.uuid4()
     cfg.LAYPA_UUID = str(_uuid)
 
-    version_path = Path("version_info")
-
     cfg.LAYPA_GIT_HASH = get_git_hash()
 
-    cfg.CONFIG_PATH = str(Path(args.config).resolve())
+    config_path = Path(args.config).resolve()
+    cfg.CONFIG_PATH = str(config_path)
 
     if not hasattr(args, "train"):
         pass
@@ -131,6 +130,9 @@ def setup_cfg(args, cfg: Optional[CfgNode] = None) -> CfgNode:
             folder_name.append(cfg.NAME)
         cfg.OUTPUT_DIR = os.path.join(cfg.OUTPUT_DIR, "_".join(folder_name))
 
+    if cfg.MODEL.RESUME:
+        cfg.OUTPUT_DIR = str(config_path.parent)
+
     # Overwrite device based on detected hardware
     if cfg.MODEL.DEVICE:
         # If CUDA device cannot be found, default to CPU