[bugfix] Fix the output format for VideoClips.subset (#6700) (#6706)

Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com> Co-authored-by: Vasilis Vryniotis <datumbox@users.noreply.github.com>
pytorch · Oct 5, 2022 · 3bd6ce9 · 3bd6ce9
1 parent 8762598
commit 3bd6ce9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchvision/datasets/video_utils.py b/torchvision/datasets/video_utils.py
@@ -198,6 +198,7 @@ def subset(self, indices: List[int]) -> "VideoClips":
             _video_max_dimension=self._video_max_dimension,
             _audio_samples=self._audio_samples,
             _audio_channels=self._audio_channels,
+            output_format=self.output_format,
         )
 
     @staticmethod