From 9152c80f392dbb2ed70bc8dd2d0c99428f8e602e Mon Sep 17 00:00:00 2001
From: xujinling <jinling.xu@zilliz.com>
Date: Wed, 22 Jun 2022 19:08:34 +0800
Subject: [PATCH] l

Signed-off-by: xujinling <jinling.xu@zilliz.com>
---
 bridge_former.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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