From 300e9f6225a7bd2d5d149569b106c14909d961da Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Mon, 1 Jun 2026 10:50:03 +0000 Subject: [PATCH 1/2] Initial plan From 955e83908cf86668c2af8f0272d75211873df6ad Mon Sep 17 00:00:00 2001 From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com> Date: Mon, 1 Jun 2026 10:55:47 +0000 Subject: [PATCH 2/2] fix: allow non-strict decoder checkpoint loading Co-authored-by: mkofler96 <18218171+mkofler96@users.noreply.github.com> --- DeepSDFStruct/deep_sdf/workspace.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/DeepSDFStruct/deep_sdf/workspace.py b/DeepSDFStruct/deep_sdf/workspace.py index 77b5418d..bb034b39 100644 --- a/DeepSDFStruct/deep_sdf/workspace.py +++ b/DeepSDFStruct/deep_sdf/workspace.py @@ -148,7 +148,7 @@ def load_model_parameters( data = torch.load(filename, map_location=device, weights_only=True) - decoder.load_state_dict(data["model_state_dict"]) + decoder.load_state_dict(data["model_state_dict"], strict=False) return data["epoch"] @@ -317,13 +317,13 @@ def load_trained_model( data = torch.load(filename, map_location=device) decoder = init_decoder(experiment_specs, device, data_parallel) try: - decoder.load_state_dict(data["model_state_dict"]) + decoder.load_state_dict(data["model_state_dict"], strict=False) except RuntimeError: state_dict = {} for k, v in data["model_state_dict"].items(): new_key = k.replace("module.", "", 1) if k.startswith("module.") else k state_dict[new_key] = v - decoder.load_state_dict(state_dict) + decoder.load_state_dict(state_dict, strict=False) decoder = decoder.to(device) return decoder