diff --git a/torchvision/datasets/kinetics.py b/torchvision/datasets/kinetics.py index 2c8faa343d6..7b52991b3cd 100644 --- a/torchvision/datasets/kinetics.py +++ b/torchvision/datasets/kinetics.py @@ -41,7 +41,6 @@ def __init__(self, root, frames_per_clip, step_between_clips=1, frame_rate=None, num_workers=1, _video_width=0, _video_height=0, _video_min_dimension=0, _audio_samples=0): super(Kinetics400, self).__init__(root) - extensions = ('avi',) classes = list(sorted(list_dir(root))) class_to_idx = {classes[i]: i for i in range(len(classes))}