From 627625803dd24cc99695d70face6ba0b747d8fc2 Mon Sep 17 00:00:00 2001 From: Yannick Gibson Date: Sun, 1 Feb 2026 12:32:25 +0100 Subject: [PATCH] Fix dtype mismatch in strided evaluation visibility assignment --- cotracker/evaluation/core/evaluator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/cotracker/evaluation/core/evaluator.py b/cotracker/evaluation/core/evaluator.py index 7b31f9d6..77092101 100644 --- a/cotracker/evaluation/core/evaluator.py +++ b/cotracker/evaluation/core/evaluator.py @@ -268,6 +268,7 @@ def evaluate_sequence( mask = pred_trj == 0 pred_trj[mask] = inv_pred_trj[mask] + pred_vsb = pred_vsb.to(inv_pred_vsb.dtype) # Ensure dtypes match before assignment pred_vsb[mask[:, :, :, 0]] = inv_pred_vsb[mask[:, :, :, 0]] pred_tracks = pred_trj, pred_vsb