Fix bug in subsample.py. Thank you to @Guangyun-Xu for finding this bug.

dcharatan · Jun 11, 2024 · cff9450 · cff9450
1 parent 19ac72b
commit cff9450
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flowmap/subsample.py b/flowmap/subsample.py
@@ -128,6 +128,7 @@ def subsample_frames(
 
         # Compute the mean flows.
         mean_flows.append(flow_predictor.forward(videos).norm(dim=-1).mean().item())
+        last = current
 
     flow_step = sum(mean_flows) / target_num_frames
     remaining = 0