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
|
B, N, D = queries.shape
|
||||||
|
|
||||||
assert D == 3
|
assert D == 3
|
||||||
assert B == 1
|
|
||||||
|
|
||||||
video = video.reshape(B * T, C, H, W)
|
video = video.reshape(B * T, C, H, W)
|
||||||
video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)
|
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,
|
backward_tracking=False,
|
||||||
):
|
):
|
||||||
B, T, C, H, W = video.shape
|
B, T, C, H, W = video.shape
|
||||||
assert B == 1
|
|
||||||
|
|
||||||
video = video.reshape(B * T, C, H, W)
|
video = video.reshape(B * T, C, H, W)
|
||||||
video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)
|
video = F.interpolate(video, tuple(self.interp_shape), mode="bilinear", align_corners=True)
|
||||||
|
Loading…
Reference in New Issue
Block a user