From 4a9286e17f59bae9f9b15219ce8aabce3f10dd5f Mon Sep 17 00:00:00 2001 From: JunkyByte Date: Tue, 25 Jul 2023 16:31:44 +0200 Subject: [PATCH] fix --- cotracker/predictor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cotracker/predictor.py b/cotracker/predictor.py index 8dcf3ca..cb0945b 100644 --- a/cotracker/predictor.py +++ b/cotracker/predictor.py @@ -133,7 +133,7 @@ class CoTrackerPredictor(torch.nn.Module): ) if add_support_grid: - grid_pts = get_points_on_a_grid(self.support_grid_size, self.interp_shape, device=self.device) + grid_pts = get_points_on_a_grid(self.support_grid_size, self.interp_shape, device=video.device) grid_pts = torch.cat( [torch.zeros_like(grid_pts[:, :, :1]), grid_pts], dim=2 )