remove assert B==1
This commit is contained in:
parent
9c9a97d158
commit
721fcc237b
@ -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)
|
||||
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user