remove assert B==1

This commit is contained in:
Nikita Karaev 2023-12-28 17:27:30 +00:00
parent 9c9a97d158
commit 721fcc237b
2 changed files with 0 additions and 2 deletions

View File

@ -38,7 +38,6 @@ class EvaluationPredictor(torch.nn.Module):
B, N, D = queries.shape
assert D == 3
assert B == 1
video = video.reshape(B * T, C, H, W)
video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)

View File

@ -92,7 +92,6 @@ class CoTrackerPredictor(torch.nn.Module):
backward_tracking=False,
):
B, T, C, H, W = video.shape
assert B == 1
video = video.reshape(B * T, C, H, W)
video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)