修改跟踪算法cpu gpu切换错误

2025-01-15 15:38:50 +08:00
parent 31db54fa37
commit ad850221c5
4 changed files with 16 additions and 11 deletions
--- a/contrast/feat_extract/inference.py
+++ b/contrast/feat_extract/inference.py
@ -108,7 +108,7 @@ class FeatsInterface:

            # patch = patch.to(device=self.device).half()
            if str(self.device) != "cpu":
-                patch = patch.to(device=self.device).half()
+                patch = patch.to(device=self.device)
            else:
                patch = patch.to(device=self.device)