This commit is contained in:
Eugene Fairley 2025-06-28 00:01:32 -07:00
parent 5feb9a6915
commit 32553f6d96

View File

@ -500,6 +500,7 @@ def patch_motion(
): ):
"""Apply motion patching based on tracks""" """Apply motion patching based on tracks"""
with torch.no_grad(): with torch.no_grad():
print("vid shape:", vid)
_, T, H, W = vid.shape _, T, H, W = vid.shape
N = tracks.shape[2] N = tracks.shape[2]
_, tracks_xy, visible = torch.split( _, tracks_xy, visible = torch.split(