Skip to content
Permalink
Browse files

fix(encoder):

fix mixture encoder
  • Loading branch information...
Larryjianfeng committed Aug 1, 2019
1 parent 95f03c5 commit 17779676bd452b2679ba7e64918721cf85e0bec1
Showing with 2 additions and 2 deletions.
  1. +2 −2 gnes/encoder/video/incep_mixture.py
@@ -106,8 +106,8 @@ def post_init(self):
def encode(self, videos: List['np.ndarray'], *args, **kwargs) -> np.ndarray:
ret = []
v_len = [len(v) for v in videos]
pos_start = [0] + [sum(v_len[:i]) for i in range(1, len(v_len)-1)]
pos_end = [sum(v_len[:i]) for i in range(len(v_len))]
pos_start = [0] + [sum(v_len[:i+1]) for i in range(len(v_len)-1)]
pos_end = [sum(v_len[:i+1]) for i in range(len(v_len))]
max_len = min(max(v_len), self.max_frames)

img = [im for v in videos for im in v]

0 comments on commit 1777967

Please sign in to comment.
You can’t perform that action at this time.