Restore CerebNet checkpoint functions

Restore FastSurfer checkpoint functions inherited by CerebNet via import. Those functions are used in downstream modules.
Deep-MI · Sep 9, 2024 · 7128010 · 7128010
1 parent 4557d3b
commit 7128010
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/CerebNet/utils/checkpoint.py b/CerebNet/utils/checkpoint.py
@@ -19,9 +19,26 @@
 if TYPE_CHECKING:
     import yacs
 
+from FastSurferCNN.utils.checkpoint import (
+    load_from_checkpoint,
+    create_checkpoint_dir,
+    get_checkpoint,
+    get_checkpoint_path,
+    save_checkpoint,
+)
 from FastSurferCNN.utils import logging
 from FastSurferCNN.utils.parser_defaults import FASTSURFER_ROOT
 
+__all__ = [
+    "create_checkpoint_dir",
+    "get_checkpoint",
+    "get_checkpoint_path",
+    "is_checkpoint_epoch",
+    "load_from_checkpoint",
+    "save_checkpoint",
+    "YAML_DEFAULT",
+]
+
 # DEFAULTS
 YAML_DEFAULT = FASTSURFER_ROOT / "CerebNet/config/checkpoint_paths.yaml"