diff --git a/bridge_former.py b/bridge_former.py index 025247b..754178c 100644 --- a/bridge_former.py +++ b/bridge_former.py @@ -61,7 +61,8 @@ class BridgeFormer(NNOperator): assert len(video.shape) == 4 video = video.transpose(3, 0, 1, 2) # thwc -> cthw - + if self.skip_preprocess: + self.transform_cfgs.update(num_frames=None) video = transform_video( video=video, **self.transform_cfgs