@@ -131,7 +131,7 @@ def forward(
131131 b , c , t , f = x .size ()
132132 x = self .out (x .transpose (1 , 2 ).contiguous ().view (b , t , c * f ))
133133 x , pos_emb = self .pos_enc (x , offset )
134- return x , pos_emb , x_mask [:, :, : - 2 : 2 ][:, :, : - 2 :2 ]
134+ return x , pos_emb , x_mask [:, :, 2 :: 2 ][:, :, 2 : :2 ]
135135
136136
137137class Conv2dSubsampling6 (BaseSubsampling ):
@@ -182,7 +182,7 @@ def forward(
182182 b , c , t , f = x .size ()
183183 x = self .linear (x .transpose (1 , 2 ).contiguous ().view (b , t , c * f ))
184184 x , pos_emb = self .pos_enc (x , offset )
185- return x , pos_emb , x_mask [:, :, : - 2 : 2 ][:, :, : - 4 :3 ]
185+ return x , pos_emb , x_mask [:, :, 2 :: 2 ][:, :, 4 : :3 ]
186186
187187
188188class Conv2dSubsampling8 (BaseSubsampling ):
@@ -237,4 +237,4 @@ def forward(
237237 b , c , t , f = x .size ()
238238 x = self .linear (x .transpose (1 , 2 ).contiguous ().view (b , t , c * f ))
239239 x , pos_emb = self .pos_enc (x , offset )
240- return x , pos_emb , x_mask [:, :, : - 2 : 2 ][:, :, : - 2 : 2 ][:, :, : - 2 :2 ]
240+ return x , pos_emb , x_mask [:, :, 2 :: 2 ][:, :, 2 :: 2 ][:, :, 2 : :2 ]
0 commit comments