From 9acb7c58f29bfbca21ed72210b57a769138912d0 Mon Sep 17 00:00:00 2001 From: Tony Franca Date: Wed, 26 Nov 2025 14:35:05 +0000 Subject: [PATCH] prefix saved checkpoint keys with `detector.` (otherwise inference breaks) --- sam3/train/trainer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/sam3/train/trainer.py b/sam3/train/trainer.py index a7ed1988..5250a551 100644 --- a/sam3/train/trainer.py +++ b/sam3/train/trainer.py @@ -355,6 +355,8 @@ def save_checkpoint(self, epoch, checkpoint_names=None): checkpoint_paths.append(os.path.join(checkpoint_folder, f"{ckpt_name}.pt")) state_dict = unwrap_ddp_if_wrapped(self.model).state_dict() + # Add 'detector.' prefix to match checkpoint format expected by model loading code + state_dict = {"detector." + k: v for k, v in state_dict.items()} state_dict = exclude_params_matching_unix_pattern( patterns=self.checkpoint_conf.skip_saving_parameters, state_dict=state_dict )