update tensor contiguous for dpt head

2025-03-20 17:42:14 +00:00
parent 497e1bade4
commit ae1b9c8750
1 changed files with 1 additions and 1 deletions
@@ -206,7 +206,7 @@ class DPTHead(nn.Module):
            Tensor or Tuple[Tensor, Tensor]: Feature maps or (predictions, confidence).
        """
        if frames_start_idx is not None and frames_end_idx is not None:
-            images = images[:, frames_start_idx:frames_end_idx]
+            images = images[:, frames_start_idx:frames_end_idx].contiguous()

        B, S, _, H, W = images.shape