add @torch.jit.script to ImageList (pytorch#1919)

fmassa · Jun 9, 2020 · 9539e1b · 9539e1b
1 parent addae9f
commit 9539e1b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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