in transforms.py [0:0]
def __init__(self, crop_size: int = 224, num_crops: int = 3):
super().__init__()
self.crop_size = crop_size
if num_crops == 3:
self.crops_to_ext = [0, 1, 2]
elif num_crops == 1:
self.crops_to_ext = [1]
else:
raise NotImplementedError(
"Nothing else supported yet, "
"slowfast only takes 0, 1, 2 as arguments"
)