diff --git a/torchvision/models/detection/image_list.py b/torchvision/models/detection/image_list.py index b1d0600e9c1..570f1f9c353 100644 --- a/torchvision/models/detection/image_list.py +++ b/torchvision/models/detection/image_list.py @@ -6,6 +6,7 @@ from torch import Tensor +@torch.jit.script class ImageList(object): """ Structure that holds a list of images (of possibly