Bugfix

probicheaux · probicheaux · commit c5c60e5fd00c · 2025-07-23T06:15:05.000-07:00
diff --git a/rfdetr/detr.py b/rfdetr/detr.py
@@ -329,7 +329,7 @@ def deploy_to_roboflow(self, workspace: str, project_ids: List[str], api_key: st
         outpath = os.path.join(tmp_out_dir, "weights.pth")
         torch.save(
             {
-                "model": self.model.model,
+                "model": self.model.model.state_dict(),
                 "args": self.model.args
             }, outpath
         )

Original file line number	Diff line number	Diff line change
`@@ -329,7 +329,7 @@ def deploy_to_roboflow(self, workspace: str, project_ids: List[str], api_key: st`
`329`	`329`	`outpath = os.path.join(tmp_out_dir, "weights.pth")`
`330`	`330`	`torch.save(`
`331`	`331`	`{`
`332`		`- "model": self.model.model,`
	`332`	`+ "model": self.model.model.state_dict(),`
`333`	`333`	`"args": self.model.args`
`334`	`334`	`}, outpath`
`335`	`335`	`)`