Upload utils.py

src/sbv2/utils.py

@@ -0,0 +1,12 @@
+import torch
+
+def load_checkpoint(filepath, model, optimizer=None, strict=True):
+    print(f"Loading checkpoint: {filepath}")
+    checkpoint_dict = torch.load(filepath, map_location="cpu")
+    model.load_state_dict(checkpoint_dict["model"], strict=strict)
+
+    if optimizer is not None and "optimizer" in checkpoint_dict:
+        optimizer.load_state_dict(checkpoint_dict["optimizer"])
+
+    print("Checkpoint loaded.")
+    return model