From ae1b9c87507ea53c111c0bc79c20fb601d6ac283 Mon Sep 17 00:00:00 2001 From: Jianyuan Wang Date: Thu, 20 Mar 2025 17:42:14 +0000 Subject: [PATCH] update tensor contiguous for dpt head --- vggt/heads/dpt_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vggt/heads/dpt_head.py b/vggt/heads/dpt_head.py index c8c8af9..fa5cf6e 100644 --- a/vggt/heads/dpt_head.py +++ b/vggt/heads/dpt_head.py @@ -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