diff --git a/torchvision/datasets/video_utils.py b/torchvision/datasets/video_utils.py index c4890ff4416..b607def243d 100644 --- 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